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

Question

我在此页面上关注此问题：https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynb

这个tut使用预先训练的模型图'frozen_inference_graph.pb'。现在我想为我的数据集

构建我的自定义模型

有一个直截了当的例子吗？我根本不想使用Google云端服务进行培训，所有操作都必须在我的本地计算机上完成

Answer 1

当然！你绝对可以在这里使用命令行进行本地训练： 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，但即​​使在本地进行培训，许多提示仍然具有相关性。

Answer 2

我跟着this guide，非常棒，几乎很容易，他展示了如何训练Mac和奶酪探测器，希望它有所帮助！