OpenCV - 檢測圖像中的多個對象（之一）

Question

我有木箱的圖像。

我必須單獨檢測每個木箱。 它看起來類似於下圖：

木箱

你對如何做到這一點有什么想法嗎？
我應該使用Al嗎？ 或者只是像 SVM 這樣的機器學習？ 或者一些模式識別算法？ 或者我可以訓練它。

訓練數據集

我試圖檢測圓/橢圓，但效果不佳。 我還讀到木材反射紅色。
但是我對 OpenCV 沒有太多經驗，所以我不知道哪種方法最適合這項任務。
感謝您的幫助

Answer 1

我認為重新訓練 YOLO 似乎是一個不錯的選擇：

https://github.com/AlexeyAB/darknet

您需要大約 2,000 張帶標簽的圖像，以及圖像增強。 我已經使用這個庫為 YOLO 進行圖像增強：

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