Unverified Commit d716669f authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #66 from deepghs/dev/classify 

dev(narugo): add generic classify method
parents 8813aa0a ee1850bd

docs/source/api_doc/validate/aicheck_benchmark.plot.py

+6 −5
Original line number Diff line number Diff line 
import random



from benchmark import BaseBenchmark, create_plot_cli

from imgutils.generic.classify import _open_models_for_repo_id

from imgutils.validate import get_ai_created_score

from imgutils.validate.aicheck import _MODEL_NAMES

from imgutils.validate.aicheck import _REPO_ID



_MODEL_NAMES = _open_models_for_repo_id(_REPO_ID).model_names





class AnimeAICheckBenchmark(BaseBenchmark):
@@ -11,12 +14,10 @@ class AnimeAICheckBenchmark(BaseBenchmark): 
        self.model = model



    def load(self):

        from imgutils.validate.aicheck import _open_anime_aicheck_model

        _ = _open_anime_aicheck_model(self.model)

        _open_models_for_repo_id(_REPO_ID)._open_model(self.model)



    def unload(self):

        from imgutils.validate.aicheck import _open_anime_aicheck_model

        _open_anime_aicheck_model.cache_clear()

        _open_models_for_repo_id(_REPO_ID).clear()



    def run(self):

        image_file = random.choice(self.all_images)

docs/source/api_doc/validate/aicheck_benchmark.plot.py.svg

+624 −480

File changed.

Preview size limit exceeded, changes collapsed.

docs/source/api_doc/validate/bangumi_char_benchmark.plot.py

+5 −14
Original line number Diff line number Diff line 
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_bangumi_char

from imgutils.validate.bangumi_char import _REPO_ID



hf_fs = HfFileSystem()



_REPOSITORY = 'deepghs/bangumi_char_type'

_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 AnimeBangumiCharacterBenchmark(BaseBenchmark):
@@ -21,12 +14,10 @@ class AnimeBangumiCharacterBenchmark(BaseBenchmark): 
        self.model = model



    def load(self):

        from imgutils.validate.bangumi_char import _open_anime_bangumi_char_model

        _ = _open_anime_bangumi_char_model(self.model)

        _open_models_for_repo_id(_REPO_ID)._open_model(self.model)



    def unload(self):

        from imgutils.validate.bangumi_char import _open_anime_bangumi_char_model

        _open_anime_bangumi_char_model.cache_clear()

        _open_models_for_repo_id(_REPO_ID).clear()



    def run(self):

        image_file = random.choice(self.all_images)

docs/source/api_doc/validate/bangumi_char_benchmark.plot.py.svg

+190 −166

File changed.

Preview size limit exceeded, changes collapsed.

docs/source/api_doc/validate/classify_benchmark.plot.py

+6 −5
Original line number Diff line number Diff line 
import random



from benchmark import BaseBenchmark, create_plot_cli

from imgutils.generic.classify import _open_models_for_repo_id

from imgutils.validate import anime_classify

from imgutils.validate.classify import _MODEL_NAMES

from imgutils.validate.classify import _REPO_ID



_MODEL_NAMES = _open_models_for_repo_id(_REPO_ID).model_names





class AnimeClassifyBenchmark(BaseBenchmark):
@@ -11,12 +14,10 @@ class AnimeClassifyBenchmark(BaseBenchmark): 
        self.model = model



    def load(self):

        from imgutils.validate.classify import _open_anime_classify_model

        _ = _open_anime_classify_model(self.model)

        _open_models_for_repo_id(_REPO_ID)._open_model(self.model)



    def unload(self):

        from imgutils.validate.classify import _open_anime_classify_model

        _open_anime_classify_model.cache_clear()

        _open_models_for_repo_id(_REPO_ID).clear()



    def run(self):

        image_file = random.choice(self.all_images)