Merge pull request #97 from deepghs/dev/eva2

dev(narugo): add support for eva02 large tagger model
deepghs · Jul 29, 2024 · 328c06f · 328c06f
2 parents 0438c23 + a6f9100
commit 328c06f
Show file tree

Hide file tree

Showing 2 changed files with 793 additions and 782 deletions.
diff --git a/docs/source/api_doc/tagging/tagging_benchmark.plot.py b/docs/source/api_doc/tagging/tagging_benchmark.plot.py
@@ -59,10 +59,11 @@ def run(self):
             ('wd14-convnextv2', Wd14Benchmark("ConvNextV2")),
             ('wd14-vit', Wd14Benchmark("ViT")),
             ('wd14-moat', Wd14Benchmark("MOAT")),
-            ('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")),
+            ('wd-swinv2-v3', Wd14Benchmark("SwinV2_v3")),
+            ('wd-vit-v3', Wd14Benchmark("ViT_v3")),
+            ('wd-convnext-v3', Wd14Benchmark("ConvNext_v3")),
+            ('wd-vit-large-tagger-v3', Wd14Benchmark("ViT_Large")),
+            ('wd-eva02-large-tagger-v3', Wd14Benchmark("EVA02_Large")),
             ('mldanbooru', MLDanbooruBenchmark()),
         ],
         title='Benchmark for Tagging Models',