Commit bf02d499 authored by Jingwei Zhang's avatar Jingwei Zhang Committed by ZwwWayne
Browse files

[Fix] update the key name of modules from mmdet in cfgs (#2011)

parent 8eeb8136
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
model = dict( model = dict(
type='CascadeRCNN', type='CascadeRCNN',
pretrained='torchvision://resnet50', pretrained='torchvision://resnet50',
_scope_='mmdet',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -63,9 +63,9 @@ model = dict( ...@@ -63,9 +63,9 @@ model = dict(
# model training and testing settings # model training and testing settings
train_cfg=dict( train_cfg=dict(
_scope_='mmdet',
img_rpn=dict( img_rpn=dict(
assigner=dict( assigner=dict(
_scope_='mmdet',
type='MaxIoUAssigner', type='MaxIoUAssigner',
pos_iou_thr=0.7, pos_iou_thr=0.7,
neg_iou_thr=0.3, neg_iou_thr=0.3,
...@@ -73,7 +73,7 @@ model = dict( ...@@ -73,7 +73,7 @@ model = dict(
match_low_quality=True, match_low_quality=True,
ignore_iof_thr=-1), ignore_iof_thr=-1),
sampler=dict( sampler=dict(
type='mmdet.RandomSampler', type='RandomSampler',
num=256, num=256,
pos_fraction=0.5, pos_fraction=0.5,
neg_pos_ub=-1, neg_pos_ub=-1,
...@@ -90,7 +90,6 @@ model = dict( ...@@ -90,7 +90,6 @@ model = dict(
min_bbox_size=0), min_bbox_size=0),
img_rcnn=dict( img_rcnn=dict(
assigner=dict( assigner=dict(
_scope_='mmdet',
type='MaxIoUAssigner', type='MaxIoUAssigner',
pos_iou_thr=0.5, pos_iou_thr=0.5,
neg_iou_thr=0.5, neg_iou_thr=0.5,
...@@ -98,7 +97,7 @@ model = dict( ...@@ -98,7 +97,7 @@ model = dict(
match_low_quality=False, match_low_quality=False,
ignore_iof_thr=-1), ignore_iof_thr=-1),
sampler=dict( sampler=dict(
type='mmdet.RandomSampler', type='RandomSampler',
num=512, num=512,
pos_fraction=0.25, pos_fraction=0.25,
neg_pos_ub=-1, neg_pos_ub=-1,
......
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
model = dict( model = dict(
type='MaskRCNN', type='MaskRCNN',
pretrained='torchvision://resnet50', pretrained='torchvision://resnet50',
_scope_='mmdet',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -6,6 +6,7 @@ _base_ = [ ...@@ -6,6 +6,7 @@ _base_ = [
model = dict( model = dict(
type='HybridTaskCascade', type='HybridTaskCascade',
pretrained='torchvision://resnet50', pretrained='torchvision://resnet50',
_scope_='mmdet',
backbone=dict( backbone=dict(
type='ResNet', type='ResNet',
depth=50, depth=50,
......
...@@ -19,7 +19,7 @@ model = dict( ...@@ -19,7 +19,7 @@ model = dict(
train_cfg=dict( train_cfg=dict(
_delete_=True, _delete_=True,
assigner=dict( assigner=dict(
type='MaxIoUAssigner', type='Max3DIoUAssigner',
iou_calculator=dict(type='BboxOverlapsNearest3D'), iou_calculator=dict(type='BboxOverlapsNearest3D'),
pos_iou_thr=0.6, pos_iou_thr=0.6,
neg_iou_thr=0.45, neg_iou_thr=0.45,
......
...@@ -174,7 +174,7 @@ model = dict( ...@@ -174,7 +174,7 @@ model = dict(
min_pos_iou=0.4, min_pos_iou=0.4,
ignore_iof_thr=-1), ignore_iof_thr=-1),
dict( # animal dict( # animal
type='MaxIoUAssigner', type='Max3DIoUAssigner',
iou_calculator=dict(type='BboxOverlapsNearest3D'), iou_calculator=dict(type='BboxOverlapsNearest3D'),
pos_iou_thr=0.55, pos_iou_thr=0.55,
neg_iou_thr=0.4, neg_iou_thr=0.4,
......
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