模型地址在pyssss.json里面 把下载下来的模型 model.onnx 改为 仓库名.onnx 把下载下来的csv配置文件 selected_tags.csv 改为 仓库名.csv 4. 模型仓库列表 根据json 文件中的信息,可知有10个模型 https://huggingface.co/SmilingWolf/wd-vit-tagger-v3 https://huggingface.co/SmilingWolf/wd-swinv2-tagger-...
"wd-v1-4-swinv2-tagger-v2":"https://huggingface.co/SmilingWolf/wd-v1-4-swin2-tagger-v2" 18- } 19- ] 2+ "name":"WD14Tagger", 3+ "logging":false, 4+ "settings": { 5+ "model":"wd-v1-4-moat-tagger-v2", 6+ "threshold":0.35, ...
"wd-v1-4-swinv2-tagger-v2":"https://huggingface.co/SmilingWolf/wd-v1-4-swinv2-tagger-v2", 21- "wd-v1-4-vit-tagger":"https://huggingface.co/SmilingWolf/wd-v1-4-vit-tagger" 13+ "wd-vit-tagger-v3":"{HF_ENDPOINT}/SmilingWolf/wd-vit-tagger-v3", ...
13 + "wd-eva02-large-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-eva02-large-tagger-v3", 13 14 "wd-vit-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-vit-tagger-v3", 14 15 "wd-swinv2-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-swinv2-tagger-v3", 15 16 "wd-convnext-tagger-...
For example, if using the repositorySmilingWolf/wd-swinv2-tagger-v3with a batch size of 4, and the training data is located in the parent foldertrain_data, it would be: python tag_images_by_wd14_tagger.py--onnx--repo_id SmilingWolf/wd-swinv2-tagger-v3--batch_size4....
python fintune/tag_images_by_wd14_tagger.py --onnx --repo_id <モデルのrepo id> --batch_size <バッチサイズ> <教師データフォルダ> ``` レポジトリに `SmilingWolf/wd-swinv2-tagger-v3` を使用し、バッチサイズを4にして、教師データを親フォルダの `train_data`に置いた場合、以...
### [`wd14-convnext-v2`](https://huggingface.co/SmilingWolf/wd-v1-4-convnext-tagger-v2) ``` animal focus, clothes writing, earrings, full body, meme, shirt, shoes, simple background, socks, solo, sweat, tail, white background, white shirt ``` ### [`wd14-swinv2-v2`](ht...