Commit 86b05cc0 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): merge from main

parents d5176ff6 09418348

.github/workflows/test.yml

+0 −4
Original line number Diff line number Diff line
@@ -17,15 +17,11 @@ jobs: 
          - 'windows-latest'

          - 'macos-latest'

        python-version:

          - '3.7'

          - '3.8'

          - '3.9'

          - '3.10'

          - '3.11'

          - '3.12'

        exclude:

          - os: 'macos-latest'

            python-version: '3.7'



    steps:

      - name: Get system version for Linux

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.