Tensorflow物体检测使用自定义模式
问题描述:
我按照此页面上此啧啧:https://github.com/tensorflow/models/blob/master/object_detection/object_detection_tutorial.ipynbTensorflow物体检测使用自定义模式
这个啧啧使用的预训练的模型图“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,但即使在本地进行培训,许多提示仍然相关。