JaMe76 commited on
Commit
2a4ac51
1 Parent(s): 86e7608

Update CASCADE_RCNN_R_50_FPN_GN.yaml

Browse files

Adding

SCORE_THRESH_TEST: 0.4
NMS_THRESH_TEST: 0.001
NMS_THRESH_CLASS_AGNOSTIC: 0.999

Files changed (1) hide show
  1. CASCADE_RCNN_R_50_FPN_GN.yaml +3 -0
CASCADE_RCNN_R_50_FPN_GN.yaml CHANGED
@@ -13,6 +13,8 @@ MODEL:
13
  ROI_HEADS:
14
  NAME: "CascadeROIHeads"
15
  NUM_CLASSES: 2
 
 
16
  ROI_BOX_HEAD:
17
  CLS_AGNOSTIC_BBOX_REG: True
18
  PIXEL_MEAN:
@@ -42,3 +44,4 @@ INPUT:
42
  - 200
43
  TEST:
44
  DETECTIONS_PER_IMAGE: 350
 
 
13
  ROI_HEADS:
14
  NAME: "CascadeROIHeads"
15
  NUM_CLASSES: 2
16
+ SCORE_THRESH_TEST: 0.4
17
+ NMS_THRESH_TEST: 0.001
18
  ROI_BOX_HEAD:
19
  CLS_AGNOSTIC_BBOX_REG: True
20
  PIXEL_MEAN:
 
44
  - 200
45
  TEST:
46
  DETECTIONS_PER_IMAGE: 350
47
+ NMS_THRESH_CLASS_AGNOSTIC: 0.999