1b87a98261
## 主要更新 - ✨ 更新所有依赖到最新稳定版本 - 📝 添加详细的项目文档和模型推荐 - 🔧 配置 VSCode Cloud Studio 预览功能 - 🐛 修复 PyTorch API 弃用警告 ## 依赖更新 - diffusers: 0.27.2 → 0.35.2 - gradio: 4.21.0 → 5.46.0 - peft: 0.7.1 → 0.18.0 - Pillow: 9.5.0 → 11.3.0 - fastapi: 0.108.0 → 0.116.2 ## 新增文件 - CLAUDE.md - 项目架构和开发指南 - UPGRADE_NOTES.md - 详细的升级说明 - .vscode/preview.yml - 预览配置 - .vscode/LAUNCH_GUIDE.md - 启动指南 - .gitignore - 更新的忽略规则 ## 代码修复 - 修复 iopaint/model/ldm.py 中的 torch.cuda.amp.autocast() 弃用警告 ## 文档更新 - README.md - 添加模型推荐和使用指南 - 完整的项目源码（iopaint/） - Web 前端源码（web_app/） 🤖 Generated with Claude Code
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
import cv2
|
|
import os
|
|
|
|
from anytext_pipeline import AnyTextPipeline
|
|
from utils import save_images
|
|
|
|
seed = 66273235
|
|
# seed_everything(seed)
|
|
|
|
pipe = AnyTextPipeline(
|
|
ckpt_path="/Users/cwq/code/github/IOPaint/iopaint/model/anytext/anytext_v1.1_fp16.ckpt",
|
|
font_path="/Users/cwq/code/github/AnyText/anytext/font/SourceHanSansSC-Medium.otf",
|
|
use_fp16=False,
|
|
device="mps",
|
|
)
|
|
|
|
img_save_folder = "SaveImages"
|
|
rgb_image = cv2.imread(
|
|
"/Users/cwq/code/github/AnyText/anytext/example_images/ref7.jpg"
|
|
)[..., ::-1]
|
|
|
|
masked_image = cv2.imread(
|
|
"/Users/cwq/code/github/AnyText/anytext/example_images/edit7.png"
|
|
)[..., ::-1]
|
|
|
|
rgb_image = cv2.resize(rgb_image, (512, 512))
|
|
masked_image = cv2.resize(masked_image, (512, 512))
|
|
|
|
# results: list of rgb ndarray
|
|
results, rtn_code, rtn_warning = pipe(
|
|
prompt='A cake with colorful characters that reads "EVERYDAY", best quality, extremely detailed,4k, HD, supper legible text, clear text edges, clear strokes, neat writing, no watermarks',
|
|
negative_prompt="low-res, bad anatomy, extra digit, fewer digits, cropped, worst quality, low quality, watermark, unreadable text, messy words, distorted text, disorganized writing, advertising picture",
|
|
image=rgb_image,
|
|
masked_image=masked_image,
|
|
num_inference_steps=20,
|
|
strength=1.0,
|
|
guidance_scale=9.0,
|
|
height=rgb_image.shape[0],
|
|
width=rgb_image.shape[1],
|
|
seed=seed,
|
|
sort_priority="y",
|
|
)
|
|
if rtn_code >= 0:
|
|
save_images(results, img_save_folder)
|
|
print(f"Done, result images are saved in: {img_save_folder}")
|