Spaces:
Running
Running
narugo1992
commited on
Commit
•
7e9b170
1
Parent(s):
49e14f2
dev(narugo): use m model
Browse files
person.py
CHANGED
|
@@ -8,6 +8,7 @@ from plot import plot_detection
|
|
| 8 |
from yolo_ import _image_preprocess, _data_simple_postprocess
|
| 9 |
|
| 10 |
_PERSON_MODELS = [
|
|
|
|
| 11 |
'person_detect_best_s.onnx',
|
| 12 |
]
|
| 13 |
_DEFAULT_PERSON_MODEL = _PERSON_MODELS[0]
|
|
|
|
| 8 |
from yolo_ import _image_preprocess, _data_simple_postprocess
|
| 9 |
|
| 10 |
_PERSON_MODELS = [
|
| 11 |
+
'person_detect_best_m.onnx',
|
| 12 |
'person_detect_best_s.onnx',
|
| 13 |
]
|
| 14 |
_DEFAULT_PERSON_MODEL = _PERSON_MODELS[0]
|