jatamura commited on
Commit
db94881
·
verified ·
1 Parent(s): dd12693

Update configs/test_model_config.yaml

Browse files
Files changed (1) hide show
  1. configs/test_model_config.yaml +3 -3
configs/test_model_config.yaml CHANGED
@@ -198,11 +198,11 @@ MODEL:
198
  IOU_THRESHOLDS:
199
  - 0.5
200
  NAME: CascadeROIHeads
201
- NMS_THRESH_TEST: 0.5
202
  NUM_CLASSES: 3
203
  POSITIVE_FRACTION: 0.8
204
  PROPOSAL_APPEND_GT: true
205
- SCORE_THRESH_TEST: 0.05
206
  ROI_KEYPOINT_HEAD:
207
  CONV_DIMS:
208
  - 512
@@ -253,7 +253,7 @@ MODEL:
253
  - 0.5
254
  - 0.7
255
  LOSS_WEIGHT: 1.0
256
- NMS_THRESH: 0.5
257
  POSITIVE_FRACTION: 0.8
258
  POST_NMS_TOPK_TEST: 4500
259
  POST_NMS_TOPK_TRAIN: 6000
 
198
  IOU_THRESHOLDS:
199
  - 0.5
200
  NAME: CascadeROIHeads
201
+ NMS_THRESH_TEST: 0.7
202
  NUM_CLASSES: 3
203
  POSITIVE_FRACTION: 0.8
204
  PROPOSAL_APPEND_GT: true
205
+ SCORE_THRESH_TEST: 0.5
206
  ROI_KEYPOINT_HEAD:
207
  CONV_DIMS:
208
  - 512
 
253
  - 0.5
254
  - 0.7
255
  LOSS_WEIGHT: 1.0
256
+ NMS_THRESH: 0.9
257
  POSITIVE_FRACTION: 0.8
258
  POST_NMS_TOPK_TEST: 4500
259
  POST_NMS_TOPK_TRAIN: 6000