| _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 | |