TensorFlow+SSD+OpenCV+python完成自训练数据的实时目标检测

该博客介绍了如何使用TensorFlow的SSD模型进行自定义目标检测,包括创建训练数据集、使用LabelImg标注、转换为TFRecord格式、配置训练参数、训练模型、导出PB模型,最后通过OpenCV在无TensorFlow环境中调用模型进行实时检测。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

介绍:利用tensorflow进行ssd实时目标检测,并实现自己的数据集训练,利用object_detection api具体步骤,并且将模型实现opencv调用,可以实现无tensorflow环境下的模型调用实时检测。本案例只采用了一类目标,可以自行增加!

1 准备图片训练集

图片命名最好是classname+num(图片格式为jpg,如果为png后面会报错)

2 利用labelimg软件进行目标边框的处理,得到xml文件

此处注意修改labelimg软件文件夹中的data文件,这是自己制作边框时可选择的label名。

选择边框之后就可以保存为xml文件了,保存的xml文件名与图片名字一一对应。

3 生成tfrecod数据格式文件

生成的xml文件全部存入xmls文件夹内,原始图片全部存入images,并把xmls文件夹放入annotations中,利用split.py文件生成四个训练验证的txt,txt文件其实就是索引图片的位置。

修改文件pbtxt,这是label_map文件,放在data中。修改dog_labe

评论 31
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值