Commit db723b01 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): update docs

parent ac7e4f2c
Loading
Loading
Loading
Loading
+3 −5
Original line number Diff line number Diff line
@@ -13,6 +13,9 @@ Overview:
    .. image:: face_detect_benchmark.plot.py.svg
        :align: center

    The models are hosted on
    `huggingface - deepghs/anime_face_detection <https://huggingface.co/deepghs/anime_face_detection>`_.

"""
from functools import lru_cache
from typing import List, Tuple
@@ -23,11 +26,6 @@ from ._yolo import _image_preprocess, _data_postprocess
from ..data import ImageTyping, load_image, rgb_encode
from ..utils import open_onnx_model

_VERSIONS = {
    'v0': '',
    'v1': 'v1_',
}


@lru_cache()
def _open_face_detect_model(level: str = 's', version: str = 'v1.3'):