Merge pull request #57 from deepghs/dev/ages

dev(narugo): init version for anime style ages
deepghs · Dec 16, 2023 · 4f95410 · 4f95410
2 parents c595725 + 7210bb2
commit 4f95410
Show file tree

Hide file tree

Showing 65 changed files with 4,110 additions and 0 deletions.
diff --git a/docs/source/api_doc/validate/index.rst b/docs/source/api_doc/validate/index.rst
@@ -16,5 +16,6 @@ imgutils.validate
     nsfw
     portrait
     rating
+    style_age
     teen
     truncate
diff --git a/docs/source/api_doc/validate/style_age.plot.py b/docs/source/api_doc/validate/style_age.plot.py
@@ -0,0 +1,18 @@
+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('style_age', '1970s-', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('style_age', '1980s', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('style_age', '1990s', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('style_age', '2000s', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('style_age', '2010s', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('style_age', '2015s', '*.jpg'))),
+        *natsorted(glob.glob(os.path.join('style_age', '2020s', '*.jpg'))),
+        columns=4, figsize=(10, 26),
+    )
diff --git a/docs/source/api_doc/validate/style_age.plot.py.svg b/docs/source/api_doc/validate/style_age.plot.py.svg
diff --git a/docs/source/api_doc/validate/style_age.rst b/docs/source/api_doc/validate/style_age.rst
@@ -0,0 +1,21 @@
+imgutils.validate.style_age
+=============================================
+
+.. currentmodule:: imgutils.validate.style_age
+
+.. automodule:: imgutils.validate.style_age
+
+
+anime_style_age_score
+-----------------------------
+
+.. autofunction:: anime_style_age_score
+
+
+
+anime_style_age
+-----------------------------
+
+.. autofunction:: anime_style_age
+
+
diff --git a/docs/source/api_doc/validate/style_age/1970s-/1.jpg b/docs/source/api_doc/validate/style_age/1970s-/1.jpg
diff --git a/docs/source/api_doc/validate/style_age/1970s-/2.jpg b/docs/source/api_doc/validate/style_age/1970s-/2.jpg
diff --git a/docs/source/api_doc/validate/style_age/1970s-/3.jpg b/docs/source/api_doc/validate/style_age/1970s-/3.jpg
diff --git a/docs/source/api_doc/validate/style_age/1970s-/4.jpg b/docs/source/api_doc/validate/style_age/1970s-/4.jpg
diff --git a/docs/source/api_doc/validate/style_age/1980s/5.jpg b/docs/source/api_doc/validate/style_age/1980s/5.jpg
diff --git a/docs/source/api_doc/validate/style_age/1980s/6.jpg b/docs/source/api_doc/validate/style_age/1980s/6.jpg
diff --git a/docs/source/api_doc/validate/style_age/1980s/7.jpg b/docs/source/api_doc/validate/style_age/1980s/7.jpg
diff --git a/docs/source/api_doc/validate/style_age/1980s/8.jpg b/docs/source/api_doc/validate/style_age/1980s/8.jpg
diff --git a/docs/source/api_doc/validate/style_age/1990s/10.jpg b/docs/source/api_doc/validate/style_age/1990s/10.jpg
diff --git a/docs/source/api_doc/validate/style_age/1990s/11.jpg b/docs/source/api_doc/validate/style_age/1990s/11.jpg
diff --git a/docs/source/api_doc/validate/style_age/1990s/12.jpg b/docs/source/api_doc/validate/style_age/1990s/12.jpg
diff --git a/docs/source/api_doc/validate/style_age/1990s/9.jpg b/docs/source/api_doc/validate/style_age/1990s/9.jpg
diff --git a/docs/source/api_doc/validate/style_age/2000s/13.jpg b/docs/source/api_doc/validate/style_age/2000s/13.jpg
diff --git a/docs/source/api_doc/validate/style_age/2000s/14.jpg b/docs/source/api_doc/validate/style_age/2000s/14.jpg
diff --git a/docs/source/api_doc/validate/style_age/2000s/15.jpg b/docs/source/api_doc/validate/style_age/2000s/15.jpg
diff --git a/docs/source/api_doc/validate/style_age/2000s/16.jpg b/docs/source/api_doc/validate/style_age/2000s/16.jpg
diff --git a/docs/source/api_doc/validate/style_age/2010s/17.jpg b/docs/source/api_doc/validate/style_age/2010s/17.jpg
diff --git a/docs/source/api_doc/validate/style_age/2010s/18.jpg b/docs/source/api_doc/validate/style_age/2010s/18.jpg
diff --git a/docs/source/api_doc/validate/style_age/2010s/19.jpg b/docs/source/api_doc/validate/style_age/2010s/19.jpg
diff --git a/docs/source/api_doc/validate/style_age/2010s/20.jpg b/docs/source/api_doc/validate/style_age/2010s/20.jpg
diff --git a/docs/source/api_doc/validate/style_age/2015s/21.jpg b/docs/source/api_doc/validate/style_age/2015s/21.jpg
diff --git a/docs/source/api_doc/validate/style_age/2015s/22.jpg b/docs/source/api_doc/validate/style_age/2015s/22.jpg
diff --git a/docs/source/api_doc/validate/style_age/2015s/23.jpg b/docs/source/api_doc/validate/style_age/2015s/23.jpg
diff --git a/docs/source/api_doc/validate/style_age/2015s/24.jpg b/docs/source/api_doc/validate/style_age/2015s/24.jpg
diff --git a/docs/source/api_doc/validate/style_age/2020s/25.jpg b/docs/source/api_doc/validate/style_age/2020s/25.jpg
diff --git a/docs/source/api_doc/validate/style_age/2020s/26.jpg b/docs/source/api_doc/validate/style_age/2020s/26.jpg
diff --git a/docs/source/api_doc/validate/style_age/2020s/27.jpg b/docs/source/api_doc/validate/style_age/2020s/27.jpg
diff --git a/docs/source/api_doc/validate/style_age/2020s/28.jpg b/docs/source/api_doc/validate/style_age/2020s/28.jpg
diff --git a/docs/source/api_doc/validate/style_age_benchmark.plot.py b/docs/source/api_doc/validate/style_age_benchmark.plot.py
@@ -0,0 +1,45 @@
+import os
+import random
+
+from huggingface_hub import HfFileSystem
+
+from benchmark import BaseBenchmark, create_plot_cli
+from imgutils.validate import anime_style_age
+
+hf_fs = HfFileSystem()
+
+_REPOSITORY = 'deepghs/anime_style_ages'
+_MODEL_NAMES = [
+    os.path.relpath(file, _REPOSITORY).split('/')[0] for file in
+    hf_fs.glob(f'{_REPOSITORY}/*/model.onnx')
+]
+
+
+class AnimeStyleAgeBenchmark(BaseBenchmark):
+    def __init__(self, model):
+        BaseBenchmark.__init__(self)
+        self.model = model
+
+    def load(self):
+        from imgutils.validate.style_age import _open_anime_style_age_model
+        _ = _open_anime_style_age_model(self.model)
+
+    def unload(self):
+        from imgutils.validate.style_age import _open_anime_style_age_model
+        _open_anime_style_age_model.cache_clear()
+
+    def run(self):
+        image_file = random.choice(self.all_images)
+        _ = anime_style_age(image_file, self.model)
+
+
+if __name__ == '__main__':
+    create_plot_cli(
+        [
+            (name, AnimeStyleAgeBenchmark(name))
+            for name in _MODEL_NAMES
+        ],
+        title='Benchmark for Anime Style Age Models',
+        run_times=10,
+        try_times=20,
+    )()