【CV】如何使用Tensorflow提供的Object Detection API--4--开始训练模型

至此已经学习了如何选择预训练模型，将数据集转为TFRecord格式。模型和数据都准备好了，是时候开启训练了。

这些在COCO数据集上的模型都是针对90类进行识别的，如果自己的任务没有这么多类，或者类不同怎么办呢？

如果是我们不是用物体检测的API的话，答案是移除最后的90个类的分类器层，替换为一个新的神经网络层。

shape = (fc_2nd_last_get_shape().as_list()[-1],nb_classes)
fc_last_W = tf.Variable(tf.truncated_normal(shape, stddev=1e-2))
fc_last_b = tf.Variable(tf.zeros(nb_classes))
logits = tf.nn.xw_plus_b(fc_2nd_last, fc_last_W, fc_last_b)

但是对于物体检测的API而言，我们只需要修改一下配置文件即可。

在object_detection/samples/configs文件夹下，有各种预训练模型的配置文件。

以Faster-RCNN举例子：

# Faster R-CNN with Inception Resnet v2, Atrous version, with Cosine
# Learning Rate schedule.
# Trained on COCO, initialized from Imagenet classification checkpoint
# Users should configure the fine_tune_checkpoint field in the train config as
# well as the label_map_path and input_path fields in the train_input_reader and
# eval_input_reader. Search for "PATH_TO_BE_CONFIGURED" to find the fields that
# should be configured.

——TBD—

参考：

https://medium.com/@WuStangDan/step-by-step-tensorflow-object-detection-api-tutorial-part-4-training-the-model-68a9e5d5a333

【CV】如何使用Tensorflow提供的Object Detection API--4--开始训练模型相关推荐

【CV】如何使用Tensorflow提供的Object Detection API --1--使用预训练模型
Tensorflow的物体检测API是非常强大的工具,它可以使得没有机器学习背景的人都可以快速部署强大的图像识别,物体检测模型.但是,考虑到使用的指导文档不够丰富,具体如何使用成了很多人的门槛. 本篇 ...
【CV】如何使用Tensorflow提供的Object Detection API --2--数据转换为TFRecord格式
本篇主要讲的是如何将现存的数据变成Tensorflow记录格式,然后我们就可以用这些数据来进行微调模型,以解决我们关心的问题了. 什么是TFRecord格式一般使用TF读取数据有四种方式: 预先把所 ...
【CV】如何使用Tensorflow提供的Object Detection API--3--手工标注数据
前面两篇看完,我们已经知道如何选用预训练模型以及将现存的其他数据集变成TFRecord格式的数据了. 但是如果需要用你自己的数据集,该怎么办呢? 本篇主要讲如何创建自己的数据集,并用object_de ...
TensorFlow Object Detection API Custom Object Hangs On
TensorFlow Object Detection API Hangs On - Training and Evaluating using Custom Object Detector *The ...
【物体检测快速入门系列 | 01 】基于Tensorflow2.x Object Detection API构建自定义物体检测器
这是机器未来的第1篇文章原文首发地址:https://blog.csdn.net/RobotFutures/article/details/124745966 CSDN话题挑战赛第1期活动详情地址 ...
Tensorflow object detection API 搭建物体识别模型
----------------------------------------------------先把别人博客教程跑通-------------------------------------- ...
谷歌开放的TensorFlow Object Detection API 效果如何？对业界有什么影响
? 谷歌开放了一个 Object Detection API: Supercharge your C 写个简单的科普帖吧. 熟悉TensorFlow的人都知道,tf在Github上的主页是:tenso ...
tensorflow精进之路(二十五)——Object Detection API目标检测(下)（VOC数据集训练自己的模型进行目标检测）
1.概述上一讲,我们使用了别人根据COCO数据集训练好的模型来做目标检测,这一讲,我们就来训练自己的模型. 2.下载数据集为了方便学习,我们先使用别人整理好的数据集来训练---VOC 2012数据 ...
tensorflow精进之路(二十四)——Object Detection API目标检测(中)（COCO数据集训练的模型—ssd_mobilenet_v1_coco模型）
1.概述上一讲简单的讲了目标检测的原理以及Tensorflow Object Detection API的安装,这一节继续讲Tensorflow Object Detection API怎么用. 2 ...

【CV】如何使用Tensorflow提供的Object Detection API--4--开始训练模型

【CV】如何使用Tensorflow提供的Object Detection API--4--开始训练模型相关推荐

最新文章

热门文章