Implement camie-tagger.

2025-03-25 20:03:38 +09:00
parent 7aa7155a8d
commit aaaa713f01
2 changed files with 85 additions and 76 deletions
@@ -1,6 +1,7 @@
 from tagger.interrogator.interrogator import AbsInterrogator
 from tagger.interrogator.waifudiffusioninterrogator import WaifuDiffusionInterrogator
 from tagger.interrogator.mldanbooruinterrogator import MLDanbooruInterrogator
+from tagger.interrogator.camietaggerinterrogator import CamieTaggerInterrogator

 interrogators: dict[str, AbsInterrogator] = {
    'wd14-vit.v1': WaifuDiffusionInterrogator(
@@ -73,4 +74,10 @@ interrogators: dict[str, AbsInterrogator] = {
        repo_id='deepghs/ml-danbooru-onnx',
        model_path='TResnet-D-FLq_ema_6-30000.onnx'
    ),
+    'camie-tagger': CamieTaggerInterrogator(
+        'Camie Tagger',
+        repo_id='Camais03/camie-tagger',
+        model_path='model_initial.onnx',
+        tags_path='model_initial_metadata.json'
+    ),
 }