OpenCV-检测图像中的多个对象（之一）

Question

我有带有[[木箱的图像。

我必须分别检测

每个

木箱。它看起来类似于下图：wooden trunks

您对如何做到这一点有任何想法吗？我应该使用铝吗？还是像SVM这样的机器学习？还是一些模式识别算法？或者我可以训练它。

training dataset

我尝试检测圆形/椭圆形，但是效果不佳。我也读过木头反射红色。但是我在OpenCV方面没有太多经验，所以我不知道哪种方法最适合此任务。谢谢您的帮助

Answer 1

我认为对YOLO进行再培训似乎是一个不错的选择：

https://github.com/AlexeyAB/darknet

您将需要大约2,000张带标签的图像，以及图像增强。我已经将此库用于YOLO的图像增强：

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