Skip to content

Commit

Permalink
[Fix] update the key name of modules from mmdet in cfgs (#2011)
Browse files Browse the repository at this point in the history
  • Loading branch information
JingweiZhang12 authored Nov 15, 2022
1 parent 59119a0 commit bd2a49d
Show file tree
Hide file tree
Showing 6 changed files with 8 additions and 6 deletions.
1 change: 1 addition & 0 deletions configs/_base_/models/cascade-mask-rcnn_r50_fpn.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
model = dict(
type='CascadeRCNN',
pretrained='torchvision://resnet50',
_scope_='mmdet',
backbone=dict(
type='ResNet',
depth=50,
Expand Down
7 changes: 3 additions & 4 deletions configs/_base_/models/imvotenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,17 +63,17 @@

# 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,
min_pos_iou=0.3,
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,
Expand All @@ -90,15 +90,14 @@
min_bbox_size=0),
img_rcnn=dict(
assigner=dict(
_scope_='mmdet',
type='MaxIoUAssigner',
pos_iou_thr=0.5,
neg_iou_thr=0.5,
min_pos_iou=0.5,
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,
Expand Down
1 change: 1 addition & 0 deletions configs/_base_/models/mask-rcnn_r50_fpn.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
model = dict(
type='MaskRCNN',
pretrained='torchvision://resnet50',
_scope_='mmdet',
backbone=dict(
type='ResNet',
depth=50,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
model = dict(
type='HybridTaskCascade',
pretrained='torchvision://resnet50',
_scope_='mmdet',
backbone=dict(
type='ResNet',
depth=50,
Expand Down
2 changes: 1 addition & 1 deletion configs/second/second_hv_secfpn_8xb6-80e_kitti-3d-car.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
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,
Expand Down
2 changes: 1 addition & 1 deletion configs/ssn/ssn_hv_secfpn_sbn-all_16xb2-2x_lyft-3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,7 @@
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,
Expand Down

0 comments on commit bd2a49d

Please sign in to comment.