Spaces:
Running
Running
narugo1992
commited on
Commit
·
ebc873b
1
Parent(s):
3107b50
dev(narugo): allow new models
Browse files
person.py
CHANGED
|
@@ -9,6 +9,8 @@ from yolo_ import _image_preprocess, _data_postprocess
|
|
| 9 |
|
| 10 |
_PERSON_MODELS = [
|
| 11 |
'person_detect_plus_v1.1_best_m.onnx',
|
|
|
|
|
|
|
| 12 |
'person_detect_plus_best_m.onnx',
|
| 13 |
'person_detect_best_m.onnx',
|
| 14 |
'person_detect_best_x.onnx',
|
|
|
|
| 9 |
|
| 10 |
_PERSON_MODELS = [
|
| 11 |
'person_detect_plus_v1.1_best_m.onnx',
|
| 12 |
+
'person_detect_plus_v1.1_best_s.onnx',
|
| 13 |
+
'person_detect_plus_v1.1_best_n.onnx',
|
| 14 |
'person_detect_plus_best_m.onnx',
|
| 15 |
'person_detect_best_m.onnx',
|
| 16 |
'person_detect_best_x.onnx',
|