如何使用Tensorflow对象检测API定义自己的模型

时间:2018-02-16 22:22:55

标签: python-3.x tensorflow machine-learning deep-learning object-detection

我有2000多个培训图像(自定义农场结构),我想使用Tensorflow定义自己的模型。我研究过(https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/defining_your_own_model.md)。这是一个非常高级的文档。任何人都可以帮我解决这个问题吗?

1 个答案:

答案 0 :(得分:2)

文档本身就是一个高级指南,但您基本上需要创建一个基于DetectionModel的类。

您需要设计并实现在DetectionModelhere中抽象定义的方法,以便创建一个模型,该模型是此抽象核心类的实例。您如何设计这些功能取决于您,但您上面链接的文档提供了一些很好的指导。