narugo1992 commited on
Commit
7e6465f
1 Parent(s): 7f0b0d1

dev(narugo): update default model and parameters

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. head.py +1 -1
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.25, label='Score Threshold')
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