mmdetection 使用自定义数据集训练 将数据集转为COCO格式

1.mmdet的安装与配置

按照官网来就好

2. 将数据转为COCO格式

3. 编写config配置文件

我想使用centernet进行训练，编写如下config脚本
改动的内容有：num_class
此外还需要改动 mmdet/datasets/coco.py

# -*- coding: utf-8 -*-
"""
Created on Thu Nov 18 11:57:13 2021

@author: sxj96
"""

_base_ = [
    '../_base_/datasets/coco_detection.py',
    '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
]

model = dict(
    type='CenterNet',
    backbone=dict(
        type='ResNet',
        depth=18,
        norm_eval=False,
        norm_cfg=dict(type='BN'),
        init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet18')),
    neck=dict(
        type='CTResNetNeck',
        in_channel=512,
        num_deconv_filters=(256, 128, 64),
        num_deconv_kernels=(4, 4, 4),
        use_dcn=True),
    bbox_head=dict(
        type='CenterNetHead',
        #num_classes=80,
        num_classes=1,
        in_channel=64,
        feat_channel=64,
        loss_center_heatmap=dict(type='GaussianFocalLoss', loss_weight=1.0),
        loss_wh=dict(type='L1Loss', loss_weight=0.1),
        loss_offset=dict(type