使用BLIP 給圖片caption 出現(xiàn)No files with extension .txt were found 錯(cuò)誤解決
首先,出現(xiàn)這個(gè)錯(cuò)誤基本是因?yàn)锽LIP的caption_weights 文件(1.7g左右)未下載成功。
方法一:
刪除整個(gè)finetune文件夾,重新在UI里點(diǎn)Caption images,有很小概率它會(huì)重新下載組件。
方法二:
1. 從https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth 網(wǎng)站下載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" "你需要打標(biāo)簽的圖片文件夾地址" --caption_weights="你下載model_large_caption.pth的路徑地址"
回車運(yùn)行即可;
注:--batch_size等參數(shù)可以自己調(diào)整。

標(biāo)簽: