Update CASCADE_RCNN_R_50_FPN_GN.yaml
Browse filesAdding
SCORE_THRESH_TEST: 0.4
NMS_THRESH_TEST: 0.001
NMS_THRESH_CLASS_AGNOSTIC: 0.999
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
|