调试M2det算法train自己的数据(VOC2007)

1.数据格式指定VOC

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

2. 删掉VOC2012

/M2Det/configs/m2det512_vgg.py:62

dataset = 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:170

self._imgpath = os.path.join('%s', 'JPEGImages', '%s.png')   

5.指定数据路径

m2det512_vgg.py:81

VOCroot = 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数据
在这里插入图片描述

  • 1
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 21
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 21
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值