pi-ux-ce/wd14-tagger-standalone
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #17 from corkborg/large-model

Browse Source 
Add support smilingwolf's large models.
This commit is contained in:
corkborg
2024-08-26 00:39:02 +09:00
committed by GitHub
parent da8baf8d8f 5fdc560e89
commit 938abdbc14
2 changed files with 26 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+18 -7
View File
@@ -47,16 +47,27 @@ python run.py --dir dir/dir
## Support Models
```
python run.py --file image.jpg --model wd14-vit.v1
python run.py --file image.jpg --model wd14-vit.v2
python run.py --file image.jpg --model wd14-convnext.v1
python run.py --file image.jpg --model wd14-convnext.v2
python run.py --file image.jpg --model wd14-convnextv2.v1
python run.py --file image.jpg --model wd14-swinv2-v1
python run.py --file image.jpg --model wd-v1-4-moat-tagger.v2
# SmilingWolf large model. (released 2024)
python run.py --file image.jpg --model wd-vit-large-tagger-v3
python run.py --file image.jpg --model wd-eva02-large-tagger-v3
# SmilingWolf v3 model. (released 2024)
python run.py --file image.jpg --model wd-v1-4-vit-tagger.v3
python run.py --file image.jpg --model wd-v1-4-convnext-tagger.v3
python run.py --file image.jpg --model wd-v1-4-swinv2-tagger.v3
# SmilingWolf v2 model. (released 2023)
python run.py --file image.jpg --model wd-v1-4-moat-tagger.v2
python run.py --file image.jpg --model wd14-vit.v2
python run.py --file image.jpg --model wd14-convnext.v2
# SmilingWolf v1 model. (released 2022)
python run.py --file image.jpg --model wd14-vit.v1
python run.py --file image.jpg --model wd14-convnext.v1
python run.py --file image.jpg --model wd14-convnextv2.v1
python run.py --file image.jpg --model wd14-swinv2-v1
# kiriyamaX model.
python run.py --file image.jpg --model mld-caformer.dec-5-97527
python run.py --file image.jpg --model mld-tresnetd.6-30000
```
tagger/interrogators.py
+8
View File
@@ -45,6 +45,14 @@ interrogators: Dict[str, Interrogator] = {
'WD14 SwinV2 v3',
repo_id='SmilingWolf/wd-swinv2-tagger-v3'
),
'wd-vit-large-tagger-v3': WaifuDiffusionInterrogator(
'WD ViT-Large Tagger v3',
repo_id='SmilingWolf/wd-vit-large-tagger-v3'
),
'wd-eva02-large-tagger-v3': WaifuDiffusionInterrogator(
'WD EVA02-Large Tagger v3',
repo_id='SmilingWolf/wd-eva02-large-tagger-v3'
),
'mld-caformer.dec-5-97527': MLDanbooruInterrogator(
'ML-Danbooru Caformer dec-5-97527',
repo_id='deepghs/ml-danbooru-onnx',