利用GPU+mmdetection框架训练自己制作的VOC格式数据集详细过程

7月份完成了这个任务,打算将整个过程记录下来,以便自己日后查看,也供大家学习交流

计算机配置:GPU:1080 Ti  HP Precision-Tower-7810

系统信息:Anaconda 3、Python 3.6Cuda 9.0、Ubuntu16.04(版本信息很重要)

一、Ubuntu16.04+cuda9.0+anaconda3安装mmdetection

0.写在前面:安装过程中,一定要牢记各个配件版本对应,在网上搜个教程,与博主版本一致的话,会更好。好多错误都是由于版本原因引起的,导致安装失败,费时费力,劳民伤财。

1.NVIDIA显卡驱动+Cuda安装+NCCL:参考这篇博客吧,也可以自己搜索,步骤都大同小异。Ubuntu16.04下cuda9.0+cudnn7.0安装指南                          ubuntu 安装 nccl(最全)

2.anaconda的安装与使用:ubuntu16.04安装和使用Anaconda3(详细)

3.安装mmdetection:   mmdetection安装教程     这篇博客也很详细安装mmdetection

二、制作自己的mmdetection并且是VOC格式的数据集

首先需要标注好的原始数据集,可以参考这篇博客,十分详细 制作自己的VOC格式数据集,同理制作COCO格式数据集方法步骤类似。

三、开始训练自己的数据,并测试

这个可以参考github上官方教程即可,CSDN也有许多详细的教程。github官方教程,readme

感谢所有参考的博客!!!

 

 

 

 

你可能感兴趣的:(利用GPU+mmdetection框架训练自己制作的VOC格式数据集详细过程)