如何将特征提取器netwrok(例如mobilenetv2)添加到tensorflow的对象检测API

时间:2019-03-04 13:28:43

标签: tensorflow object-detection-api

This tutorial讨论了如何在Tensorflow上使用对象检测API。 我正在寻找说明如何将诸如mobilenetV2之类的特征提取器添加到tensorflow的对象检测框架的教程。

1 个答案:

答案 0 :(得分:1)

您是否已检查过Tensorflow提供的Model Zoo? :)

它包括具有各种特征提取器的各种对象检测模型,例如MobileNet,Inception,ResNet等。

在下面,您将找到Tensorflow检测模型动物园的链接,您可以在其中选择检测模型架构,基于区域的(R-CNN)或单发检测器(SSD)模型以及特征提取器。

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

您可以下载基于COCO,Kitti和Open-Images等的预训练模型的冻结图。