欢迎光临散文网 会员登陆 & 注册

使用BLIP 给图片caption 出现No files with extension .txt were found 错误解决

2023-02-27 15:34 作者:子倾之黎  | 我要投稿

首先,出现这个错误基本是因为BLIP的caption_weights 文件(1.7g左右)未下载成功。

方法一:

删除整个finetune文件夹,重新在UI里点Caption images,有很小概率它会重新下载组件。

方法二:

1. 从https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth 网站下载model_large_caption.pth 至本地;

2. 在kohya_ss文件夹根目录下打开powershell,输入:

.\venv\Scripts\python.exe "finetune/make_captions.py" --batch_size="1" --num_beams="1" --top_p="0.9" --max_length="75" --min_length="5" --caption_extension=".txt" "你需要打标签的图片文件夹地址" --caption_weights="你下载model_large_caption.pth的路径地址"

回车运行即可;

注:--batch_size等参数可以自己调整。


使用BLIP 给图片caption 出现No files with extension .txt were found 错误解决的评论 (共 条)

分享到微博请遵守国家法律