Commit 96fe00ff authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): rebuild images for docs

parent 176b66be
Loading
Loading
Loading
Loading
+31 −12
Original line number Diff line number Diff line
import random

from benchmark import BaseBenchmark, create_plot_cli
from imgutils.detect import detect_faces
from imgutils.detect.visual import detection_visualize
from plot import image_plot


def _detect(img, **kwargs):
    return detection_visualize(img, detect_faces(img, **kwargs))
class FaceDetectBenchmark(BaseBenchmark):
    def __init__(self, level, version):
        BaseBenchmark.__init__(self)
        self.level = level
        self.version = version

    def load(self):
        from imgutils.detect.face import _open_face_detect_model
        _ = _open_face_detect_model(level=self.level, version=self.version)

    def unload(self):
        from imgutils.detect.face import _open_face_detect_model
        _open_face_detect_model.cache_clear()

    def run(self):
        image_file = random.choice(self.all_images)
        _ = detect_faces(image_file, level=self.level, version=self.version)


if __name__ == '__main__':
    image_plot(
        (_detect('nian.png'), 'large scale'),
        (_detect('two_bikini_girls.png'), 'closed faces'),
        (_detect('genshin_post.jpg'), 'multiple'),
        (_detect('mostima_post.jpg'), 'anime style'),
        columns=2,
        figsize=(12, 9),
    )
    create_plot_cli(
        [
            ('face v1 (yolov8s)', FaceDetectBenchmark('s', 'v1')),
            ('face v1 (yolov8n)', FaceDetectBenchmark('n', 'v1')),
            ('face v0 (yolov8s)', FaceDetectBenchmark('s', 'v0')),
            ('face v0 (yolov8n)', FaceDetectBenchmark('n', 'v0')),
        ],
        title='Benchmark for Anime Face Detections',
        run_times=10,
        try_times=20,
    )()
+0 −526

File deleted.

Preview size limit exceeded, changes collapsed.

+28 −12
Original line number Diff line number Diff line
import random

from benchmark import BaseBenchmark, create_plot_cli
from imgutils.detect import detect_heads
from imgutils.detect.visual import detection_visualize
from plot import image_plot


def _detect(img, **kwargs):
    return detection_visualize(img, detect_heads(img, **kwargs))
class HeadDetectBenchmark(BaseBenchmark):
    def __init__(self, level):
        BaseBenchmark.__init__(self)
        self.level = level

    def load(self):
        from imgutils.detect.head import _open_head_detect_model
        _ = _open_head_detect_model(level=self.level)

    def unload(self):
        from imgutils.detect.head import _open_head_detect_model
        _open_head_detect_model.cache_clear()

    def run(self):
        image_file = random.choice(self.all_images)
        _ = detect_heads(image_file, level=self.level)


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),
    )
    create_plot_cli(
        [
            ('head (yolov8s)', HeadDetectBenchmark('s')),
            ('head (yolov8n)', HeadDetectBenchmark('n')),
        ],
        title='Benchmark for Anime Head Detections',
        run_times=10,
        try_times=20,
    )()
+0 −524

File deleted.

Preview size limit exceeded, changes collapsed.

+34 −12
Original line number Diff line number Diff line
import random

from benchmark import BaseBenchmark, create_plot_cli
from imgutils.detect import detect_person
from imgutils.detect.visual import detection_visualize
from plot import image_plot


def _detect(img, **kwargs):
    return detection_visualize(img, detect_person(img, **kwargs))
class PersonDetectBenchmark(BaseBenchmark):
    def __init__(self, level, version):
        BaseBenchmark.__init__(self)
        self.level = level
        self.version = version

    def load(self):
        from imgutils.detect.person import _open_person_detect_model
        _ = _open_person_detect_model(level=self.level, version=self.version)

    def unload(self):
        from imgutils.detect.person import _open_person_detect_model
        _open_person_detect_model.cache_clear()

    def run(self):
        image_file = random.choice(self.all_images)
        _ = detect_person(image_file, level=self.level, version=self.version)


if __name__ == '__main__':
    image_plot(
        (_detect('nian.png'), 'large scale'),
        (_detect('two_bikini_girls.png'), 'closed faces'),
        (_detect('genshin_post.jpg'), 'multiple'),
        (_detect('soldiers.jpg'), 'multiple++'),
        columns=2,
        figsize=(12, 9),
    )
    create_plot_cli(
        [
            ('person v1.1 (yolov8m)', PersonDetectBenchmark('m', 'v1.1')),
            ('person v1.1 (yolov8s)', PersonDetectBenchmark('s', 'v1.1')),
            ('person v1.1 (yolov8n)', PersonDetectBenchmark('n', 'v1.1')),
            ('person v1 (yolov8m)', PersonDetectBenchmark('m', 'v1')),
            ('person v0 (yolov8s)', PersonDetectBenchmark('s', 'v0')),
            ('person v0 (yolov8m)', PersonDetectBenchmark('m', 'v0')),
            ('person v0 (yolov8x)', PersonDetectBenchmark('x', 'v0')),
        ],
        title='Benchmark for Anime Person Detections',
        run_times=10,
        try_times=20,
    )()
Loading