Commit 84d168e3 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add eyes detections

parent 5644cfb7

docs/source/api_doc/detect/eye.rst

0 → 100644
+14 −0
Original line number Diff line number Diff line 
imgutils.detect.eye

==========================



.. currentmodule:: imgutils.detect.eye



.. automodule:: imgutils.detect.eye





detect_eyes

------------------------------



.. autofunction:: detect_eyes

docs/source/api_doc/detect/eye_detect_benchmark.plot.py

0 → 100644
+37 −0
Original line number Diff line number Diff line 
import random



from benchmark import BaseBenchmark, create_plot_cli

from imgutils.detect import detect_eyes





class EyeDetectBenchmark(BaseBenchmark):

    def __init__(self, level, version):

        BaseBenchmark.__init__(self)

        self.level = level

        self.version = version



    def load(self):

        from imgutils.detect.eye import _open_eye_detect_model

        _ = _open_eye_detect_model(level=self.level, version=self.version)



    def unload(self):

        from imgutils.detect.eye import _open_eye_detect_model

        _open_eye_detect_model.cache_clear()



    def run(self):

        image_file = random.choice(self.all_images)

        _ = detect_eyes(image_file, level=self.level, version=self.version)





if __name__ == '__main__':

    create_plot_cli(

        [

            ('eye v1.0 (yolov8s)', EyeDetectBenchmark('s', 'v1.0')),

            ('eye v1.0 (yolov8n)', EyeDetectBenchmark('n', 'v1.0')),

            ('eye v0.8 (yolov8s)', EyeDetectBenchmark('s', 'v0.8')),

            ('eye v0.7 (yolov8s)', EyeDetectBenchmark('s', 'v0.7')),

        ],

        title='Benchmark for Anime Eyes Detections',

        run_times=10,

        try_times=20,

    )()

docs/source/api_doc/detect/eye_detect_demo.plot.py

0 → 100644
+19 −0
Original line number Diff line number Diff line 
from imgutils.detect import detect_eyes

from imgutils.detect.eye import _LABELS

from imgutils.detect.visual import detection_visualize

from plot import image_plot





def _detect(img, **kwargs):

    return detection_visualize(img, detect_eyes(img, **kwargs), _LABELS)





if __name__ == '__main__':

    image_plot(

        (_detect('nian.png'), 'large scale'),

        (_detect('two_bikini_girls.png'), 'closed heads'),

        (_detect('genshin_post.jpg'), 'multiple'),

        (_detect('mostima_post.jpg'), 'anime style'),

        columns=2,

        figsize=(12, 9),

    )

docs/source/api_doc/detect/index.rst

+1 −0
Original line number Diff line number Diff line
@@ -10,6 +10,7 @@ imgutils.detect 
    :maxdepth: 3



    censor

    eye

    face

    halfbody

    hand

imgutils/detect/__init__.py

+1 −0
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ Overview: 
        :align: center

"""

from .censor import detect_censors

from .eye import detect_eyes

from .face import detect_faces

from .halfbody import detect_halfbody

from .hand import detect_hands