faster-rcnn_r50-caffe_fpn_ms-1x_coco-person.py 582 B

1234567891011121314
  1. _base_ = './faster-rcnn_r50-caffe_fpn_ms-1x_coco.py'
  2. model = dict(roi_head=dict(bbox_head=dict(num_classes=1)))
  3. metainfo = {
  4. 'classes': ('person', ),
  5. 'palette': [
  6. (220, 20, 60),
  7. ]
  8. }
  9. train_dataloader = dict(dataset=dict(metainfo=metainfo))
  10. val_dataloader = dict(dataset=dict(metainfo=metainfo))
  11. test_dataloader = dict(dataset=dict(metainfo=metainfo))
  12. load_from = 'https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_3x_coco/faster_rcnn_r50_caffe_fpn_mstrain_3x_coco_bbox_mAP-0.398_20200504_163323-30042637.pth' # noqa