machine-learning - 使用自定义模型进行 Tensorflow 对象检测

Question

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

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

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

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在本例中，他使用了云，但即使在本地进行培训，许多技巧仍然是相关的。