SSD数据集的准备

在对SSD模型进行微调的时候需要重新准备数据集,简单梳理一下数据集的准备流程,大致可以分为以下几步:1.数据标注常用的数据标注工具有labelImg、ImageLabel、Yolo_mark 、BBox-Label-Tool,在此使用的是labelimg2.生成txt文件在1中对数据标注过后会有对应的xml文件生成,需要将他们转化成txt文件,在此使用的是python语言自动完成(...
摘要由CSDN通过智能技术生成

在对SSD模型进行微调的时候需要重新准备数据集,简单梳理一下数据集的准备流程,大致可以分为以下几步:

1.数据标注

常用的数据标注工具有labelImgImageLabelYolo_mark BBox-Label-Tool,在此使用的是labelimg

2.生成txt文件

在1中对数据标注过后会有对应的xml文件生成,需要将他们转化成txt文件,在此使用的是python语言自动完成(需要注意路径)

import os
import random 
#先划分训练集+验证集(0.8)和测试集,再将训练集+验证集划分开(一样一半)
trainval_percent = 0.8 
train_percent = 0.7 
xmlfilepath = 'C:\\Users\\asus\\Desktop\\Annotations' 
txtsavepath = 'C:\\Users\\asus\\Desktop\\ImageSets\\Main' 
total_xml = os.listdir(xmlfilepath) 

num=len(total_xml) 
tv=int(num*trainval_percent)#训练集+验证集
tr=int(tv*train_percent) #训练集
trainval= random.sample
  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值