centernet-update_r18_fpn_8xb8-amp-lsj-200e_coco.py 244 Bytes
Newer Older
dengjb's avatar
dengjb committed
1
2
3
4
5
6
7
_base_ = './centernet-update_r50_fpn_8xb8-amp-lsj-200e_coco.py'

model = dict(
    backbone=dict(
        depth=18,
        init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet18')),
    neck=dict(in_channels=[64, 128, 256, 512]))