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

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

被折叠的 条评论
为什么被折叠?



