制作pascal_voc数据集:将txt、csv文件转化为xml文件的格式-----制成annotations标注

现在关于目标的检测,以及一些的比赛关于目标检测的。我觉得一开始想的模型基本都是faster-rcnn吧,毕竟是一个经典模型。如果追求速度的话,就是yolo了,求更好的性能就是SSD了。基本就是这样,而这些模型最基本的训练集的形式都是pascal_voc数据集。我们要想训练自己的数据集。有两种办法:一种是修改数据接口,将代码与自己数据集的形式结合起来。比较麻烦,修改的东西挺多。我不是很建议这种方法。第二种就是我要讲的。把自己的数据集制作成VOC数据集的形式。主要制作的是annotations的文件。

我这里以csv-----xml形式来解释(txt差不多):

1.读取csv文件。

with open(filename) as f:
    reader = csv.reader(f)
    head_row = next(reader)
    for row in reader:

逐行读取文件,每一行的我通过打印最终的格式为:

e3d5c101-6bb9-48af-ac80-08c73304456b.jpg,173_179_162_180;508_197_157_194

第一个是图片名字。后面有分号隔开了,其实就是两个目标的框框坐标(格式为x,y,w,h)

2.那么重点来了,如何将上述得到的结果转化为一个对应的xml文件呢?

    	num = label.count(';')
 
  • 0
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值