Unverified Commit dc3776d0 authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #96 from deepghs/dev/vitlarge 

dev(narugo): add support for wd-vit-large-tagger-v3
parents 09fd11ae 5b6a714e

docs/source/api_doc/tagging/tagging_benchmark.plot.py

+1 −0
Original line number Diff line number Diff line
@@ -62,6 +62,7 @@ if __name__ == '__main__': 
            ('wd14-swinv2-v3', Wd14Benchmark("SwinV2_v3")),

            ('wd14-vit-v3', Wd14Benchmark("ViT_v3")),

            ('wd14-convnext-v3', Wd14Benchmark("ConvNext_v3")),

            ('wd14-vit-large-tagger-v3', Wd14Benchmark("ViT_Large")),

            ('mldanbooru', MLDanbooruBenchmark()),

        ],

        title='Benchmark for Tagging Models',

docs/source/api_doc/tagging/tagging_benchmark.plot.py.svg

+915 −821

File changed.

Preview size limit exceeded, changes collapsed.

requirements.txt

+2 −2
Original line number Diff line number Diff line 
hbutils>=0.9.0

pillow

numpy

numpy<2

scikit-learn

huggingface_hub

tqdm
@@ -8,7 +8,7 @@ opencv-python 
opencv-contrib-python

pandas

scipy

emoji>=2.5.0

emoji>=2.5.0,<2.12

pilmoji>=1.3.0

shapely

pyclipper