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