我正在为当地农场建造一个奶牛探测器。对于出现在图像中的每头奶牛(每张图像中的奶牛数量不同),我都有一个带有边界框(不是分割多边形)的图像数据集。用于完成此任务的最佳和最简单的模型是什么?许多奶牛站在其他奶牛后面,所以会有很多闭塞的情况。是否有一个我可以在 Keras 中快速运行以在该数据集上表现良好的首选模型?任何提示将不胜感激。
我应该使用什么模型进行边界框检测?
数据挖掘
喀拉斯
计算机视觉
物体检测
2022-03-13 20:21:04
1个回答
有两种非常流行的物体检测器:
由 Viola 和 Jones 介绍的Haar Cascade Classifier:在 OpenCV(Python、Java 和 C++)和 Matlab 计算机视觉工具箱(可能还有许多其他语言)上可用,是在不能选择深度学习时的一个很好的模型。
Yolo,你只看一次:它是一个实时对象检测器和分类器,它使用 DL CNN,如果我没记错的话,它的训练数据集上有奶牛(所以你可以使用迁移学习)。它是 Darknet 库的一部分,您应该完全检查该站点。另外,检查 Tiny Yolo 是否需要更快的速度。
您还应该看看:
我的建议是直接使用 YOLO 并在没有针对 COW 课程的培训的情况下进行尝试,然后通过使用您的数据重新训练几层来进行一些微调。如果你需要比这更高的准确性,你可以选择 R-CNN(可能有一些模型已经用牛训练过)。
如果您缺乏计算能力,您可以尝试 MMOD 和 HCC。但是如果你有时间你应该检查一下BDNN: Binary convolution neural networks for fast object detection,不知道它是否已经在任何库中实现,因为这篇论文相对较新(2019 年 3 月)。
其它你可能感兴趣的问题