Commit a2435a2c authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): use new model of onnx

parent 4da52aa8

docs/source/api_doc/detect/person_detect.benchmark.py

+1 −0
Original line number Diff line number Diff line
@@ -26,6 +26,7 @@ class PersonDetectBenchmark(BaseBenchmark): 
if __name__ == '__main__':

    create_plot_cli(

        [

            ('person v1.1 (yolov8m)', PersonDetectBenchmark('m', 'v1.1')),

            ('person v1 (yolov8m)', PersonDetectBenchmark('m', 'v1')),

            ('person v0 (yolov8s)', PersonDetectBenchmark('s', 'v0')),

            ('person v0 (yolov8m)', PersonDetectBenchmark('m', 'v0')),

docs/source/api_doc/detect/person_detect.benchmark.py.svg

deleted100644 → 0
+0 −2477

File deleted.

Preview size limit exceeded, changes collapsed.

imgutils/detect/person.py

+1 −1
Original line number Diff line number Diff line
@@ -37,7 +37,7 @@ def _open_person_detect_model(level: str, version: str): 
    ))





def detect_person(image: ImageTyping, level: str = 'm', version: str = 'v1', max_infer_size=640,

def detect_person(image: ImageTyping, level: str = 'm', version: str = 'v1.1', max_infer_size=640,

                  conf_threshold: float = 0.3, iou_threshold: float = 0.5):

    """

    Overview:

test/detect/test_person.py

+5 −5
Original line number Diff line number Diff line
@@ -25,11 +25,11 @@ class TestDetectPerson: 
            values.append((bbox, int(score * 1000) / 1000))



        assert values == pytest.approx([

            ((1, 141, 417, 720), 0.882),

            ((707, 5, 1264, 720), 0.83),

            ((617, 412, 801, 681), 0.773),

            ((376, 234, 558, 661), 0.75),

            ((715, 8, 1268, 720), 0.861),

            ((57, 140, 407, 720), 0.856),

            ((614, 417, 802, 680), 0.802),

            ((373, 234, 561, 657), 0.782)

        ])



    def test_detect_person_none(self):

        assert detect_person(get_testfile('png_full.png')) == []
 
        assert detect_person(get_testfile('png_full.png'), conf_threshold=0.5) == []