使用BLIP 给图片caption 出现No files with extension .txt were found 错误解决
首先,出现这个错误基本是因为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等参数可以自己调整。
