mmdet训练中数据集导入

1.引用定义的数据集类型

_base_ = ['../_base_/datasets/SIXray.py', '../_base_/default_runtime.py']

2.对于数据集中,需要修改的部分,在py文件中进行修改。


data = dict(
    samples_per_gpu=8,
    workers_per_gpu=2,
    train=dict(
        type='RepeatDataset',
        times=1,
        dataset=dict(
            pipeline=train_pipeline)),
    val=dict(
        pipeline=test_pipeline),
    test=dict(
        pipeline=test_pipeline))

数据集中,路径引用信息都在'../_base_/datasets/SIXray.py中已配置好,不需再次说明

核心要调整的就是batch(并行数据量),mmdet中对应的为 samples_per_gpu=8,单卡下表示batch为8

其次为训练数据是否重复扩充,即type='RepeatDataset',   times=1。这里表示不需要重复扩充。

你可能感兴趣的:(centernet,mmdet,深度学习,python,开发语言)