Spaces:
Running
Running
narugo1992
commited on
Commit
•
d1b96ae
1
Parent(s):
75b6d64
dev(narugo): add person detection x
Browse files
person.py
CHANGED
@@ -8,8 +8,8 @@ from plot import detection_visualize
|
|
8 |
from yolo_ import _image_preprocess, _data_postprocess
|
9 |
|
10 |
_PERSON_MODELS = [
|
11 |
-
'person_detect_best_x.onnx',
|
12 |
'person_detect_best_m.onnx',
|
|
|
13 |
'person_detect_best_s.onnx',
|
14 |
]
|
15 |
_DEFAULT_PERSON_MODEL = _PERSON_MODELS[0]
|
|
|
8 |
from yolo_ import _image_preprocess, _data_postprocess
|
9 |
|
10 |
_PERSON_MODELS = [
|
|
|
11 |
'person_detect_best_m.onnx',
|
12 |
+
'person_detect_best_x.onnx',
|
13 |
'person_detect_best_s.onnx',
|
14 |
]
|
15 |
_DEFAULT_PERSON_MODEL = _PERSON_MODELS[0]
|