mmdetection训练自己的VOC数据集

最新的请移步我的另一篇博文【2.6.0】：https://blog.csdn.net/jy1023408440/article/details/110135699

官方步骤：

https://github.com/open-mmlab/mmdetection/blob/master/GETTING_STARTED.md

首先根据规范的VOC数据集导入到项目目录下，如下图所示：

mmdetection
├── mmdet
├── tools
├── configs
├── data
│ ├── VOCdevkit
│ │ ├── VOC2007
│ │ │ ├── Annotations
│ │ │ ├── JPEGImages
│ │ │ ├── ImageSets
│ │ │ │ ├── Main
│ │ │ │ │ ├── test.txt
│ │ │ │ │ ├── trainval.txt
然后复制configs/retinanet_x101_64x4d_fpn_1x.py ,更名为my.py

修改其中dataset settings部分：

dataset_type = 'VOCDataset'
data_root = 'data/VOCdevkit/'
img_norm_cfg = dict(
mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
data = dict(
imgs_per_gpu=2,
workers_per_gpu=2,
train=dict(
type='RepeatDataset', # to avoid reloading datasets frequently
times=3,
dataset=dict(
type=dataset_type,
ann_file=[
data_root + 'VOC2007/ImageSets/Main/train.txt',#train.txt
],
img_prefix=[data_root + 'VOC2007/'],
img_scale=(1333, 800),
img_norm_cfg=img_norm_cfg,
size_divisor=32,
flip_ratio=0.5,
with_mask=False,
with_crowd=False,
with_label=True)),
val=dict(
type=dataset_type,
ann_file=data_root + 'VOC2007/ImageSets/Main/trainval.txt',
img_prefix=data_root + 'VOC2007/',
img_scale=(1333, 800),
img_norm_cfg=img_norm_cfg,
size_divisor=32,
flip_ratio=0,
with_mask=False,
with_crowd=False,
with_label=True),
test=dict(
type=dataset_type,
ann_file=data_root + 'VOC2007/ImageSets/Main/test.txt',
img_prefix=data_root + 'VOC2007/',
img_scale=(1333, 800),
img_norm_cfg=img_norm_cfg,
size_divisor=32,
flip_ratio=0,
with_mask=False,
with_crowd=False,
with_label=False,
test_mode=True))

修改mmdetection/mmdet/datasets/voc.py下classes为自己的类

如遇到label=self.cat2label 报错，请查看本人的其他博客

运行python setup.py install，重新编译

运行代码python tools/train.py config/my.py

推荐一个讲解比较详细的源代码的私人博客（如有侵权，请联系删除）
https://heary.cn/posts/mmdetection-基于PyTorch的开源目标检测系统/

mmdetection训练自己的VOC数据集相关推荐

mmdetection训练自己的VOC数据集及常见问题
训练自己的COCO数据集及常见问题见下文: mmdetection训练自己的COCO数据集及常见问题_不瘦8斤的妥球球饼的博客-CSDN博客目录一.环境安装二.训练测试步骤三.常见问题 bat ...
mmdetection的安装并训练自己的VOC数据集
mmdetection的安装并训练自己的VOC数据集 mmdetection的安装与VOC数据集的训练一. mmdetection的安装 1.使用conda创建虚拟环境 2.安装Cython 3.安 ...
在服务器上利用mmdetection来训练自己的voc数据集
在服务器上利用mmdetection来训练自己的voc数据集服务器上配置mmdetection环境在服务器上用anaconda配置自己的环境进入自己的虚拟环境,开始配置mmdetection 跑 ...
mmdetection训练自己的COCO数据集及常见问题
训练自己的VOC数据集及常见问题见下文: mmdetection训练自己的VOC数据集及常见问题_不瘦8斤的妥球球饼的博客-CSDN博客_mmdetection训练voc 目录一.环境安装二.训练 ...
mmdetectionV2.x版本训练自己的VOC数据集
mmdetectionV2.x版本训练自己的VOC数据集 1 首先根据规范的VOC数据集导入到项目目录下,如下图所示: mmdetection ----mmdet ----tools ----con ...
yolov5-7.0训练自己的VOC数据集
yolov5-7.0训练自己的VOC数据集这个笔记可能只适用于7.0版本的,写这个笔记主要是给工作室伙伴参考的,大佬请绕行有错误之处欢迎指出一.下载 yolov5的GitHub仓库地址:Rele ...
【mmdetection】SSD训练自己的VOC数据集注意事项
一.修改voc.py和class_names.py ./mmdet/datasets/voc.py 此处代码改为自己数据集的类别,以及./mmdet/core/evaluation/class_na ...
在Colab平台上用mmdetection中的Cascade RCNN训练自己的voc数据集
最近在学习目标检测,要用Cascde RCNN训练自己的数据集,因为本地电脑没有GPU,所以就借用Colab云端来进行模型训练,想要记录一下操作过程,以及希望可以给更多电脑没有GPU,但是需要用mmd ...
【学习记录】win10搭建YOLOX训练自己的VOC数据集
我是目录: 前言: 1.yolox的训练配置 2.yolox源码 3.必要的环境配置yolox所需环境 1.安装依赖库 2.安装yolox 3.安装apex 4.下载预训练模型 4.准备自己的数据集 ...

mmdetection训练自己的VOC数据集

mmdetection训练自己的VOC数据集相关推荐

最新文章

热门文章