mmdetection小技巧(强烈推荐 逐步完善中)

  • 小技巧一:在mmdetection中所有的配置文件,都有这样一行代码,如果要做目标检测,则引用coco_detection.py,如果要做目标检测,则引用coco_instance.py。在训练Swin Transformer的时候,这里的修改尤为常见,具体看你需要做什么,这里我们是做目标检测。
    _base_ = [
        '../_base_/models/mask_rcnn_r50_fpn.py',
        '../_base_/datasets/coco_detection.py', 
        '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
    ]
  • 小技巧二:在coco_detection.py中,有一个参量data_root,该参量是存放我们数据集的路径。
    dataset_type = 'CocoDataset'
    data_root = 'data/coco/'
    img_norm_cfg = dict(mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)

猜你喜欢

转载自blog.csdn.net/qq_42308217/article/details/123182981