1.數據格式指定VOC

/M2Det/train.py:20parser.add_argument('-d', '--dataset', default='VOC', help='VOC or COCO dataset')

2. 刪掉VOC2012

/M2Det/configs/m2det512_vgg.py:62dataset = dict(VOC = dict(train_sets = [('2007', 'trainval')],eval_sets = [('2007', 'test')],),# VOC = dict(# train_sets = [('2007', 'trainval'), ('2012', 'trainval')],# eval_sets = [('2007', 'test')],# ),

3.更改數據的label類型

/M2Det/data/voc0712.py:27# VOC_CLASSES = ( '__background__', # always index 0 # 'aeroplane', 'bicycle', 'bird', 'boat', # 'bottle', 'bus', 'car', 'cat', 'chair', # 'cow', 'diningtable', 'dog', 'horse', # 'motorbike', 'person', 'pottedplant', # 'sheep', 'sofa', 'train', 'tvmonitor')

4.指定圖片格式

/M2Det/data/voc0712.py:170self._imgpath = os.path.join('%s', 'JPEGImages', '%s.png')

5.指定數據路徑

m2det512_vgg.py:81VOCroot = os.path.join(home,"data/VOCdevkit/")

6.batch_size相關參數

/M2Det/configs/m2det512_vgg.py:30

7.下載 vgg16_reducedfc.pth權值文件

M2Det/configs/m2det512_vgg.py:5地址：https://github.com/amdegroot/ssd.pytorch/issues/170

8.RTX2080 +pytorch+cuda報錯問題

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : in Traceback (most recent call last)

這個問題暫時未解決(回頭解決了，補充上來)，換1080顯卡調試，沒有問題。

成功開始訓練自己的voc2007數據

總結

以上是生活随笔為你收集整理的调试M2det算法train自己的数据（VOC2007）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。