Spaces:
Sleeping
Sleeping
Update configs/test_model_config.yaml
Browse files
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.
|
| 202 |
NUM_CLASSES: 3
|
| 203 |
POSITIVE_FRACTION: 0.8
|
| 204 |
PROPOSAL_APPEND_GT: true
|
| 205 |
-
SCORE_THRESH_TEST: 0.
|
| 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.
|
| 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
|