ubuntu16.04中gpu版本caffe-ssd模型训练自己的数据

前天跑了一下caffe-ssd模型,并在pascal voc上跑了两天,效果不错,今天突然发现caffe-ssd目录没有了,,好像莫名被删除了,于是早上重新装了一遍,下午开始摸索如何用caff-ssd跑自己的数据,一个下午时间总算摸索出来了,记录一下,方便有需要的朋友参考.

caffe-ssd训练自己的数据其实挺简单的,生成目标检测的lmdb数据时候需要根据caffe-ssd中的代码生成,下面讲解一下如何操作:

需求说明:使用caffe-ssd训练文本区域检测的ssd模型,简单说就是检测一张纸上面的文字区域,检测类别只有一类textarea.(如果你有多个检测类别也可以,一样的操作方法)

1,数据准备:不管你的数据是用什么标注工具标的,肯定可以得到一个.txt文件吧(自己写一个程序把标注工具标注的文件解析出来),.txt文件里面如下第一行所示:(xmin, ymin,xmax, ymax)

data/data_ori/IMG_20171113_114301.jpg 924 740 2593 3549 0(我只有一个类别,所有标签是0)

data/data_ori/IMG_20171113_114301.jpg 332 484 1913 3261 0 1076 1036 2669 3785 1(如果你是多个类别,那一张图片肯定对应如这行一样的多个标签和坐标值)

数据先任意放在一个目录处理即可,然后写个程序该图片名字,改成和VOC一样的命名,改成数字开头即可,比如从0000000

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

洛希极限-cz

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值