diff --git a/README.md b/README.md index b57402f..1a23ed7 100644 --- a/README.md +++ b/README.md @@ -12,8 +12,7 @@ pip install -r requirements.txt ## usage ``` -usage: run.py [-h] (--dir DIR | --file FILE) [--threshold THRESHOLD] [--ext EXT] [--overwrite] [--cpu] [--rawtag] [--recursive] [--exclude-tag t1,t2,t3] - [--model {wd14-vit.v1,wd14-vit.v2,wd14-convnext.v1,wd14-convnext.v2,wd14-convnextv2.v1,wd14-swinv2-v1,wd-v1-4-moat-tagger.v2,wd-v1-4-vit-tagger.v3,wd-v1-4-convnext-tagger.v3,wd-v1-4-swinv2-tagger.v3,mld-caformer.dec-5-97527,mld-tresnetd.6-30000}] +usage: run.py [-h] (--dir DIR | --file FILE) [--threshold THRESHOLD] [--ext EXT] [--overwrite] [--cpu] [--rawtag] [--recursive] [--exclude-tag t1,t2,t3] [--model MODELNAME] options: -h, --help show this help message and exit @@ -28,8 +27,7 @@ options: --recursive Enable recursive file search --exclude-tag t1,t2,t3 Specify tags to exclude (Need comma-separated list) - --model {wd14-vit.v1,wd14-vit.v2,wd14-convnext.v1,wd14-convnext.v2,wd14-convnextv2.v1,wd14-swinv2-v1,wd-v1-4-moat-tagger.v2,wd-v1-4-vit-tagger.v3,wd-v1-4-convnext-tagger.v3,wd-v1-4-swinv2-tagger.v3,mld-caformer.dec-5-97527,mld-tresnetd.6-30000} - modelname to use for prediction (default is wd14-convnextv2.v1) + --model MODELNAME modelname to use for prediction (default is wd14-convnextv2.v1) ``` single file @@ -47,33 +45,36 @@ python run.py --dir dir/dir ## Support Models ``` +# Camie Tagger (released 2025) +python run.py --model camie-tagger --file image.jpg + # 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 +python run.py --model wd-vit-large-tagger-v3 --file image.jpg +python run.py --model wd-eva02-large-tagger-v3 --file image.jpg # 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 +python run.py --model wd-v1-4-vit-tagger.v3 --file image.jpg +python run.py --model wd-v1-4-convnext-tagger.v3 --file image.jpg +python run.py --model wd-v1-4-swinv2-tagger.v3 --file image.jpg # 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 +python run.py --model wd-v1-4-moat-tagger.v2 --file image.jpg +python run.py --model wd14-vit.v2 --file image.jpg +python run.py --model wd14-convnext.v2 --file image.jpg # 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 +python run.py --model wd14-vit.v1 --file image.jpg +python run.py --model wd14-convnext.v1 --file image.jpg +python run.py --model wd14-convnextv2.v1 --file image.jpg +python run.py --model wd14-swinv2-v1 --file image.jpg # Z3D-E621-Convnext -python run.py --file image.jpg --model z3d-e621-convnext-toynya -python run.py --file image.jpg --model z3d-e621-convnext-silveroxides +python run.py --model z3d-e621-convnext-toynya --file image.jpg +python run.py --model z3d-e621-convnext-silveroxides --file image.jpg # 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 +python run.py --model mld-caformer.dec-5-97527 --file image.jpg +python run.py --model mld-tresnetd.6-30000 --file image.jpg ``` ## Using GPU diff --git a/run.py b/run.py index a0a1818..1cc9026 100644 --- a/run.py +++ b/run.py @@ -48,7 +48,7 @@ parser.add_argument( parser.add_argument( '--model', default='wd14-convnextv2.v1', - choices=list(interrogators.keys()), + metavar='MODELNAME', help='modelname to use for prediction (default is wd14-convnextv2.v1)') args = parser.parse_args()