Commit 364698b6 authored by Cao Yuhang's avatar Cao Yuhang Committed by Kai Chen
Browse files

Remove keep all stage code in HTC and Cascade RCNN (#1806)

* Remove keep all stage code

* remove keep_all_stage in config
parent 4357697a
...@@ -174,8 +174,7 @@ test_cfg = dict( ...@@ -174,8 +174,7 @@ test_cfg = dict(
score_thr=0.05, score_thr=0.05,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -176,8 +176,7 @@ test_cfg = dict( ...@@ -176,8 +176,7 @@ test_cfg = dict(
score_thr=0.05, score_thr=0.05,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -174,8 +174,7 @@ test_cfg = dict( ...@@ -174,8 +174,7 @@ test_cfg = dict(
score_thr=0.05, score_thr=0.05,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -176,8 +176,7 @@ test_cfg = dict( ...@@ -176,8 +176,7 @@ test_cfg = dict(
score_thr=0.05, score_thr=0.05,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -176,8 +176,7 @@ test_cfg = dict( ...@@ -176,8 +176,7 @@ test_cfg = dict(
score_thr=0.05, score_thr=0.05,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -155,8 +155,7 @@ test_cfg = dict( ...@@ -155,8 +155,7 @@ test_cfg = dict(
nms_thr=0.7, nms_thr=0.7,
min_bbox_size=0), min_bbox_size=0),
rcnn=dict( rcnn=dict(
score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100), score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -164,8 +164,7 @@ test_cfg = dict( ...@@ -164,8 +164,7 @@ test_cfg = dict(
nms_thr=0.7, nms_thr=0.7,
min_bbox_size=0), min_bbox_size=0),
rcnn=dict( rcnn=dict(
score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100), score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -155,8 +155,7 @@ test_cfg = dict( ...@@ -155,8 +155,7 @@ test_cfg = dict(
nms_thr=0.7, nms_thr=0.7,
min_bbox_size=0), min_bbox_size=0),
rcnn=dict( rcnn=dict(
score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100), score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -157,8 +157,7 @@ test_cfg = dict( ...@@ -157,8 +157,7 @@ test_cfg = dict(
nms_thr=0.7, nms_thr=0.7,
min_bbox_size=0), min_bbox_size=0),
rcnn=dict( rcnn=dict(
score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100), score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -157,8 +157,7 @@ test_cfg = dict( ...@@ -157,8 +157,7 @@ test_cfg = dict(
nms_thr=0.7, nms_thr=0.7,
min_bbox_size=0), min_bbox_size=0),
rcnn=dict( rcnn=dict(
score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100), score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -177,8 +177,7 @@ test_cfg = dict( ...@@ -177,8 +177,7 @@ test_cfg = dict(
score_thr=0.05, score_thr=0.05,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -158,8 +158,7 @@ test_cfg = dict( ...@@ -158,8 +158,7 @@ test_cfg = dict(
nms_thr=0.7, nms_thr=0.7,
min_bbox_size=0), min_bbox_size=0),
rcnn=dict( rcnn=dict(
score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100), score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -190,8 +190,7 @@ test_cfg = dict( ...@@ -190,8 +190,7 @@ test_cfg = dict(
score_thr=0.05, score_thr=0.05,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -171,8 +171,7 @@ test_cfg = dict( ...@@ -171,8 +171,7 @@ test_cfg = dict(
nms_thr=0.7, nms_thr=0.7,
min_bbox_size=0), min_bbox_size=0),
rcnn=dict( rcnn=dict(
score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100), score_thr=0.05, nms=dict(type='nms', iou_thr=0.5), max_per_img=100))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -207,8 +207,7 @@ test_cfg = dict( ...@@ -207,8 +207,7 @@ test_cfg = dict(
score_thr=0.001, score_thr=0.001,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -199,8 +199,7 @@ test_cfg = dict( ...@@ -199,8 +199,7 @@ test_cfg = dict(
score_thr=0.001, score_thr=0.001,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -191,8 +191,7 @@ test_cfg = dict( ...@@ -191,8 +191,7 @@ test_cfg = dict(
score_thr=0.001, score_thr=0.001,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -191,8 +191,7 @@ test_cfg = dict( ...@@ -191,8 +191,7 @@ test_cfg = dict(
score_thr=0.001, score_thr=0.001,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -191,8 +191,7 @@ test_cfg = dict( ...@@ -191,8 +191,7 @@ test_cfg = dict(
score_thr=0.001, score_thr=0.001,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
...@@ -176,8 +176,7 @@ test_cfg = dict( ...@@ -176,8 +176,7 @@ test_cfg = dict(
score_thr=0.001, score_thr=0.001,
nms=dict(type='nms', iou_thr=0.5), nms=dict(type='nms', iou_thr=0.5),
max_per_img=100, max_per_img=100,
mask_thr_binary=0.5), mask_thr_binary=0.5))
keep_all_stages=False)
# dataset settings # dataset settings
dataset_type = 'CocoDataset' dataset_type = 'CocoDataset'
data_root = 'data/coco/' data_root = 'data/coco/'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment