Spaces:
Running
Running
narugo1992
commited on
Commit
•
7e6465f
1
Parent(s):
7f0b0d1
dev(narugo): update default model and parameters
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ if __name__ == '__main__':
|
|
42 |
gr_head_infer_size = gr.Slider(480, 960, value=640, step=32, label='Max Infer Size')
|
43 |
with gr.Row():
|
44 |
gr_head_iou_threshold = gr.Slider(0.0, 1.0, 0.7, label='IOU Threshold')
|
45 |
-
gr_head_score_threshold = gr.Slider(0.0, 1.0, 0.
|
46 |
|
47 |
gr_head_submit = gr.Button(value='Submit', variant='primary')
|
48 |
|
|
|
42 |
gr_head_infer_size = gr.Slider(480, 960, value=640, step=32, label='Max Infer Size')
|
43 |
with gr.Row():
|
44 |
gr_head_iou_threshold = gr.Slider(0.0, 1.0, 0.7, label='IOU Threshold')
|
45 |
+
gr_head_score_threshold = gr.Slider(0.0, 1.0, 0.3, label='Score Threshold')
|
46 |
|
47 |
gr_head_submit = gr.Button(value='Submit', variant='primary')
|
48 |
|
head.py
CHANGED
@@ -9,8 +9,8 @@ from plot import detection_visualize
|
|
9 |
from yolo_ import _image_preprocess, _data_postprocess
|
10 |
|
11 |
_HEAD_MODELS = [
|
12 |
-
'head_detect_best_n.onnx',
|
13 |
'head_detect_best_s.onnx',
|
|
|
14 |
]
|
15 |
_DEFAULT_HEAD_MODEL = _HEAD_MODELS[0]
|
16 |
|
|
|
9 |
from yolo_ import _image_preprocess, _data_postprocess
|
10 |
|
11 |
_HEAD_MODELS = [
|
|
|
12 |
'head_detect_best_s.onnx',
|
13 |
+
'head_detect_best_n.onnx',
|
14 |
]
|
15 |
_DEFAULT_HEAD_MODEL = _HEAD_MODELS[0]
|
16 |
|