- 首页 > it技术 > >
视觉检测图像分割干货|mmdetection小技巧(逐步完善中)
小技巧mmdetectionpytorch
- 小技巧一：在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)
推荐阅读
