Merge pull request #70 from deepghs/dev/completeness

dev(narugo): add completeness model
deepghs · Feb 26, 2024 · 452934c · 452934c
2 parents d6d9210 + 8f0e2d6
commit 452934c
Show file tree

Hide file tree

Showing 36 changed files with 4,296 additions and 0 deletions.
diff --git a/docs/source/api_doc/validate/completeness.plot.py b/docs/source/api_doc/validate/completeness.plot.py
@@ -0,0 +1,14 @@
+import glob
+import os.path
+
+from natsort import natsorted
+
+from plot import image_plot
+
+if __name__ == '__main__':
+    image_plot(
+        *natsorted(glob.glob(os.path.join('completeness', 'monochrome', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('completeness', 'rough', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('completeness', 'polished', '*.jpg'))),
+        columns=3, figsize=(8, 9),
+    )
diff --git a/docs/source/api_doc/validate/completeness.plot.py.svg b/docs/source/api_doc/validate/completeness.plot.py.svg
diff --git a/docs/source/api_doc/validate/completeness.rst b/docs/source/api_doc/validate/completeness.rst
@@ -0,0 +1,21 @@
+imgutils.validate.completeness
+=============================================
+
+.. currentmodule:: imgutils.validate.completeness
+
+.. automodule:: imgutils.validate.completeness
+
+
+anime_completeness_score
+-----------------------------
+
+.. autofunction:: anime_completeness_score
+
+
+
+anime_completeness
+-----------------------------
+
+.. autofunction:: anime_completeness
+
+
diff --git a/docs/source/api_doc/validate/completeness/monochrome/1.jpg b/docs/source/api_doc/validate/completeness/monochrome/1.jpg
diff --git a/docs/source/api_doc/validate/completeness/monochrome/2.jpg b/docs/source/api_doc/validate/completeness/monochrome/2.jpg
diff --git a/docs/source/api_doc/validate/completeness/monochrome/3.jpg b/docs/source/api_doc/validate/completeness/monochrome/3.jpg
diff --git a/docs/source/api_doc/validate/completeness/polished/7.jpg b/docs/source/api_doc/validate/completeness/polished/7.jpg
diff --git a/docs/source/api_doc/validate/completeness/polished/8.jpg b/docs/source/api_doc/validate/completeness/polished/8.jpg
diff --git a/docs/source/api_doc/validate/completeness/polished/9.jpg b/docs/source/api_doc/validate/completeness/polished/9.jpg
diff --git a/docs/source/api_doc/validate/completeness/rough/4.jpg b/docs/source/api_doc/validate/completeness/rough/4.jpg
diff --git a/docs/source/api_doc/validate/completeness/rough/5.jpg b/docs/source/api_doc/validate/completeness/rough/5.jpg
diff --git a/docs/source/api_doc/validate/completeness/rough/6.jpg b/docs/source/api_doc/validate/completeness/rough/6.jpg
diff --git a/docs/source/api_doc/validate/completeness_benchmark.plot.py b/docs/source/api_doc/validate/completeness_benchmark.plot.py
@@ -0,0 +1,36 @@
+import random
+
+from benchmark import BaseBenchmark, create_plot_cli
+from imgutils.generic.classify import _open_models_for_repo_id
+from imgutils.validate import anime_completeness
+from imgutils.validate.completeness import _REPO_ID
+
+_MODEL_NAMES = _open_models_for_repo_id(_REPO_ID).model_names
+
+
+class AnimeCompletenessBenchmark(BaseBenchmark):
+    def __init__(self, model):
+        BaseBenchmark.__init__(self)
+        self.model = model
+
+    def load(self):
+        _open_models_for_repo_id(_REPO_ID)._open_model(self.model)
+
+    def unload(self):
+        _open_models_for_repo_id(_REPO_ID).clear()
+
+    def run(self):
+        image_file = random.choice(self.all_images)
+        _ = anime_completeness(image_file, self.model)
+
+
+if __name__ == '__main__':
+    create_plot_cli(
+        [
+            (name, AnimeCompletenessBenchmark(name))
+            for name in _MODEL_NAMES
+        ],
+        title='Benchmark for Anime Completeness Models',
+        run_times=10,
+        try_times=20,
+    )()