tf2 进行目标检测训练

本文记录了使用TensorFlow 2.x进行目标检测的实践过程,包括从TF1.x到TF2.x的升级,数据集转换,预训练模型的选择与配置修改,训练过程中的参数调整,以及训练后的模型保存。在训练过程中遇到了文件路径问题、内存占用过高以及训练信息输出频率等问题,并逐一解决。

tf从0到tf1.x,现在已是tf2.x了,目标检测训练也得跟上新时代。

object detection api 配置基本差不多,数据集仍为VOC2007格式。

根据要求把文件放到各自目录后。先把数据集转成训练需要的train.record和val.record格式。

create_pascal_tf_record.py

开始是ubuntu20.04系统,python3.8,一堆错误。改为anaconda环境。

 arroplan_val.txt?? 什么原因??No such file or directory.

查看源文件,怎么有aeroplane的前缀?是bug吗?注释。

 下载ssd_mobilenet_v2_fpnlite_640*640_coco17_tpu-8.tar.gz预训练模型。

修改pileline.config文件。

修改num_classes类别数。

 修改文件路径。模型前缀为ckpt-0

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值