OpenCV - 检测图像中的多个对象(其中一个)



我有木干的图像。

我必须单独检测每个木箱。它看起来像下图所示:

木箱

你对如何做到这一点的方法有什么想法吗?
我应该使用铝吗?或者只是像 SVM 这样的机器学习?还是某种模式识别算法? 或者我可以训练它。

训练数据集

我试图检测圆圈/椭圆,但没有很好的结果。 我还读到木材反射红色。
但是我对OpenCV没有太多经验,所以我不知道哪种方法最适合这项任务。
谢谢你的帮助

我认为重新训练YOLO似乎是一个不错的选择:

https://github.com/AlexeyAB/darknet

您将需要大约 2,000 个标记图像,以及图像增强。我已经使用这个库为 YOLO 进行了图像增强:

https://github.com/aleju/imgaug/

最新更新