使用自定义模型的 Tensorflow 对象检测

Tensorflow Object Detection With Custom Model

我在这个页面上关注这个:https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb

此图使用预训练模型图 'frozen_inference_graph.pb'。现在我想为我的数据集构建自定义模型

有没有一个简单的例子可以做到这一点?我根本不想使用 Google 云服务进行培训,所有操作都必须在我的本地计算机上完成

好的!您绝对可以在此处使用命令行在本地进行训练: https://github.com/tensorflow/models/blob/master/object_detection/g3doc/running_locally.md

如果您在本地训练,我建议您使用 GPU。有人还在这里写了一篇关于如何在自定义数据集上进行训练的很好的指南:https://medium.com/towards-data-science/how-to-train-your-own-object-detector-with-tensorflows-object-detector-api-bec72ecfe1d9 在本例中,他使用了 Cloud,但即​​使在本地进行训练,许多提示仍然有用。

我已经关注了 this guide,非常棒而且实际上很简单,他展示了如何训练 Mac 和奶酪检测器,希望对您有所帮助!