diff --git a/configs/pascal_voc/faster_rcnn_r101_fpn_1x_voc0712.py b/configs/pascal_voc/faster_rcnn_r101_fpn_1x_voc0712.py index fc8391bfbdef4dcde7fd9e9134a5f851e16aa09a..546b0dd0d0f42560f3acd72ce7c937a4c619427f 100644 --- a/configs/pascal_voc/faster_rcnn_r101_fpn_1x_voc0712.py +++ b/configs/pascal_voc/faster_rcnn_r101_fpn_1x_voc0712.py @@ -1,20 +1,7 @@ -_base_ = [ - '../_base_/models/faster_rcnn_r50_fpn.py', '../_base_/datasets/voc0712.py', - '../_base_/default_runtime.py' -] +_base_ = './faster_rcnn_r50_fpn_1x_voc0712.py' model = dict( backbone=dict( depth=101, init_cfg=dict(type='Pretrained', - checkpoint='torchvision://resnet101')), - roi_head=dict(bbox_head=dict(num_classes=20))) + checkpoint='torchvision://resnet101'))) -# optimizer -optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) -optimizer_config = dict(grad_clip=None) -# learning policy -# actual epoch = 3 * 3 = 9 -lr_config = dict(policy='step', step=[3]) -# runtime settings -runner = dict( - type='EpochBasedRunner', max_epochs=4) # actual epoch = 4 * 3 = 12 diff --git a/configs/pascal_voc/faster_rcnn_r18_fpn_1x_voc0712.py b/configs/pascal_voc/faster_rcnn_r18_fpn_1x_voc0712.py new file mode 100644 index 0000000000000000000000000000000000000000..f5779d6f5f87ed7e10e92950b3b5f85ad1598fa9 --- /dev/null +++ b/configs/pascal_voc/faster_rcnn_r18_fpn_1x_voc0712.py @@ -0,0 +1,11 @@ +_base_ = './faster_rcnn_r50_fpn_1x_voc0712.py' +model = dict( + backbone=dict( + depth=18, + init_cfg=dict(type='Pretrained', + checkpoint='torchvision://resnet18')), + neck=dict( + in_channels=[64, 128, 256, 512], + )) + +