From c2487d921fbc0fe7988ff1cfd27dd1eb16fae5d1 Mon Sep 17 00:00:00 2001 From: narugo1992 Date: Tue, 9 Jan 2024 12:47:38 +0800 Subject: [PATCH] dev(narugo): regenerate doc --- .../api_doc/validate/real_benchmark.plot.py | 19 +- .../validate/real_benchmark.plot.py.svg | 2296 ----------------- 2 files changed, 5 insertions(+), 2310 deletions(-) delete mode 100644 docs/source/api_doc/validate/real_benchmark.plot.py.svg diff --git a/docs/source/api_doc/validate/real_benchmark.plot.py b/docs/source/api_doc/validate/real_benchmark.plot.py index 8975e83a7ba..3572031d2f8 100644 --- a/docs/source/api_doc/validate/real_benchmark.plot.py +++ b/docs/source/api_doc/validate/real_benchmark.plot.py @@ -1,18 +1,11 @@ -import os import random -from huggingface_hub import HfFileSystem - from benchmark import BaseBenchmark, create_plot_cli +from imgutils.generic.classify import _open_models_for_repo_id from imgutils.validate import anime_real +from imgutils.validate.real import _REPO_ID -hf_fs = HfFileSystem() - -_REPOSITORY = 'deepghs/anime_real_cls' -_MODEL_NAMES = [ - os.path.relpath(file, _REPOSITORY).split('/')[0] for file in - hf_fs.glob(f'{_REPOSITORY}/*/model.onnx') -] +_MODEL_NAMES = _open_models_for_repo_id(_REPO_ID).model_names class AnimeRealBenchmark(BaseBenchmark): @@ -21,12 +14,10 @@ def __init__(self, model): self.model = model def load(self): - from imgutils.validate.real import _open_anime_real_model - _ = _open_anime_real_model(self.model) + _open_models_for_repo_id(_REPO_ID)._open_model(self.model) def unload(self): - from imgutils.validate.real import _open_anime_real_model - _open_anime_real_model.cache_clear() + _open_models_for_repo_id(_REPO_ID).clear() def run(self): image_file = random.choice(self.all_images) diff --git a/docs/source/api_doc/validate/real_benchmark.plot.py.svg b/docs/source/api_doc/validate/real_benchmark.plot.py.svg deleted file mode 100644 index 6cacfa121e8..00000000000 --- a/docs/source/api_doc/validate/real_benchmark.plot.py.svg +++ /dev/null @@ -1,2296 +0,0 @@ - - - - - - - - 2023-12-16T10:54:00.694168 - image/svg+xml - - - Matplotlib v3.7.4, https://matplotlib.org/ - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -