使用tensorflow预训练模型进行物体识别(2)
1、概述在上一篇我们已经成功的安装了tensorflow与tensorflow 物体检测的API。下面我们将实现使用自定义数据集训练自己的对象检测模型。完成此类工作大概需要以下6个步骤1. 组织工程文件2. 组织训练数据集与标注文件3. 转化训练集为tf_record格式4. 配置训练流程 pipeline5. 监控模型训练过程6. 保存模型参数2、组织工程文件2.1 新建workspace文件夹用于存储所有的工程文件workspace 文件夹在object-detect文件夹下
原创
2021-04-16 10:31:21 ·
1441 阅读 ·
4 评论