File size: 744 Bytes
4973b58 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
_base_ = "mmdet::rtmdet/rtmdet_m_8xb32-300e_coco.py"
checkpoint = "https://download.openmmlab.com/mmdetection/v3.0/rtmdet/cspnext_rsb_pretrain/cspnext-m_8xb256-rsb-a1-600e_in1k-ecb3bbd9.pth"
model = {
"backbone": {
"init_cfg": {"type": "Pretrained", "prefix": "backbone.", "checkpoint": checkpoint},
},
"bbox_head": {"num_classes": 1},
"test_cfg": {
"nms_pre": 1000,
"min_bbox_size": 0,
"score_thr": 0.05,
"nms": {"type": "nms", "iou_threshold": 0.6},
"max_per_img": 100,
},
}
train_dataloader = {"dataset": {"metainfo": {"classes": ("person",)}}}
val_dataloader = {"dataset": {"metainfo": {"classes": ("person",)}}}
test_dataloader = val_dataloader
|