Commit bc49a1a6 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): use new infer_size

parent 4496fd30
Loading
Loading
Loading
Loading
+0 −2299

File deleted.

Preview size limit exceeded, changes collapsed.

+8 −8
Original line number Diff line number Diff line
@@ -5,14 +5,14 @@ from imgutils.detect import detect_person


class PersonDetectBenchmark(BaseBenchmark):
    def __init__(self, level, plus):
    def __init__(self, level, version):
        BaseBenchmark.__init__(self)
        self.level = level
        self.plus = plus
        self.version = version

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

    def unload(self):
        from imgutils.detect.person import _open_person_detect_model
@@ -20,16 +20,16 @@ class PersonDetectBenchmark(BaseBenchmark):

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


if __name__ == '__main__':
    create_plot_cli(
        [
            ('person plus (yolov8m)', PersonDetectBenchmark('m', True)),
            ('person (yolov8s)', PersonDetectBenchmark('s', False)),
            ('person (yolov8m)', PersonDetectBenchmark('m', False)),
            ('person (yolov8x)', PersonDetectBenchmark('x', False)),
            ('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,
+0 −2442

File deleted.

Preview size limit exceeded, changes collapsed.

+1 −1
Original line number Diff line number Diff line
@@ -12,7 +12,7 @@ if __name__ == '__main__':
        (_detect('nian.png'), 'large scale'),
        (_detect('two_bikini_girls.png'), 'closed faces'),
        (_detect('genshin_post.jpg'), 'multiple'),
        (_detect('mostima_post.jpg'), 'anime style'),
        (_detect('soldiers.jpg'), 'multiple++'),
        save_as='person_detect.dat.svg',
        columns=2,
        figsize=(12, 9),
+78.9 KiB
Loading image diff...
Loading