为什么要讲这部分呢,主要是因为我自己在这个部分也躺了很多坑。才发现这里面的工作都是需要我们自己取配置,编写相应的代码的。
目录
配置部分在configs/_base_/datasets目录下,具体实现在mmdet/datasets目录下。
1.configs
先说配置文件,以这个为例吧。https://github.com/open-mmlab/mmdetection/blob/master/configs/_base_/datasets/coco_detection.py
# dataset settings
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)
train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations', with_bbox=True),
dict(type='Resize', img_scal