对象检测和对象计数

时间:2017-12-15 22:26:45

标签: tensorflow object-detection object-detection-api

我想弄清楚如何计算图像中的对象。我需要计算鸟巢和鸟屋里的鸡蛋数量。

我想在tensorflow中使用对象检测api,然后重新训练最后一层以识别蛋。问题是我已经有很多标记的图像,但只有关于总蛋数的信息,而不是鸡蛋所在的位置。

是否有可能通过为图像提供相应的蛋计数来重新训练网络?或者我是否需要手动标记个别鸡蛋的位置?

我正在考虑将其解决为一个分类问题,其中不同的鸡蛋数量将是不同的类别(0,1,2,...... - 总共不超过8个鸡蛋)。你觉得怎么样?

1 个答案:

答案 0 :(得分:0)

最后,我选择在Tensorflow中使用对象检测api,因此我不得不在很多图像中手动标记单个鸡蛋的位置,以重新训练神经网络。结果出奇的好。

如果有人遇到相同的问题,这是我的解决方案:https://github.com/havluj/EggDetector