模型地址在 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-tagge...
For example, if using the repository SmilingWolf/wd-swinv2-tagger-v3 with a batch size of 4, and the training data is located in the parent folder train_data, it would be: python tag_images_by_wd14_tagger.py --onnx --repo_id SmilingWolf/wd-swinv2-tagger-v3 --bat...
"wd-v1-4-swinv2-tagger-v2": "https://huggingface.co/SmilingWolf/wd-v1-4-swinv2-tagger-v2", "wd-v1-4-vit-tagger": "https://huggingface.co/SmilingWolf/wd-v1-4-vit-tagger" "wd-vit-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-vit-tagger-v3", "wd-swinv2-tagger-v3": "{HF_EN...
python fintune/tag_images_by_wd14_tagger.py --onnx --repo_id <モデルのrepo id> --batch_size <バッチサイズ> <教師データフォルダ> ``` レポジトリに `SmilingWolf/wd-swinv2-tagger-v3` を使用し、バッチサイズを4にして、教師データを親フォルダの `train_data`に置いた場合、以...