out_size -> output_size

Signed-off-by: lizz <lizz@sensetime.com>
pull/2/head
lizz 2021-04-06 13:22:33 +08:00
parent 72f04b2ea1
commit 5ace2048ea
1 changed files with 2 additions and 2 deletions

View File

@ -37,7 +37,7 @@ model = dict(
type='StandardRoIHead',
bbox_roi_extractor=dict(
type='SingleRoIExtractor',
roi_layer=dict(type='RoIAlign', out_size=7, sample_num=0),
roi_layer=dict(type='RoIAlign', output_size=7, sample_num=0),
out_channels=256,
featmap_strides=[4, 8, 16, 32]),
bbox_head=dict(
@ -56,7 +56,7 @@ model = dict(
loss_bbox=dict(type='L1Loss', loss_weight=1.0)),
mask_roi_extractor=dict(
type='SingleRoIExtractor',
roi_layer=dict(type='RoIAlign', out_size=14, sample_num=0),
roi_layer=dict(type='RoIAlign', output_size=14, sample_num=0),
out_channels=256,
featmap_strides=[4, 8, 16, 32]),
mask_head=dict(