Commit 7e9008b0 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): trans n and s model to onnx

parent 39bd6c1e

zoo/person_detect.py

+3 −1
Original line number Diff line number Diff line
@@ -27,7 +27,9 @@ _KNOWN_CKPTS: List[str] = [ 
    # 'person_detect_best_m.pt',

    # 'person_detect_best_x.pt',

    # 'person_detect_plus_best_m.pt',

    'person_detect_plus_v1.1_best_m.pt',

    # 'person_detect_plus_v1.1_best_m.pt',

    'person_detect_plus_v1.1_best_s.pt',

    'person_detect_plus_v1.1_best_n.pt',

]