MxNet-SSD训练自己的数据

1、制作数据集

按照VOC格式制作了数据集

 

2、生成lst和rec文件

2.1 生成lst文件

python prepare_dataset.py --dataset pascal --year 2007 --set trainval --target ~/mxnet-ssd/data/train.lst

用于生成 train.lst文件

python prepare_dataset.py --dataset pascal --year 2007 --set test --target ~/mxnet-ssd/data/val.lst --shuffle False

 用于生成val.lst

这是数据集的目录形式,我自己制作的2012里面什么也没有,所以语句里面也不体现

生成的文件

具体内容

 

 

------------------------------------------------------------

分割线内不可用,生成的lst文件不完整,不包含目标信息。这是尝试时的操作,留下以供复习

python ~/ssd/mxnet/tools/im2rec.py --list True --recursive True --train-ratio 0.9 trail /home/king/ssd/data/VOCdevkit/JPEGImages

第一段是im2rec的路径,list表示生成lst文件,recursive True遍历root下所有文件文件夹,选练数据占90%,trail是指生成的文件的名字,最后一个是图片所在位置

------------------------------------------------------------------------------

 

2.2 生成rec文件

 python /home/king/mxnet-ssd/mxnet/tools/im2rec.py /home/king/mxnet-ssd/data /home/king/mxnet-ssd/data/VOCdevkit --pack-label 1

 

(修改中)

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值