共享相同预测特征的补丁的标签边界框

问题描述

我运行了一个基于补丁的卷积神经网络,我想将具有相同预测类别的相邻补丁“变形”为更大的补丁。在下面的示例表中,farmland 在所有其他类中的概率最高。我想创建一个包围所有这 6 个补丁的边界框,并用 farmland 标记该框。

enter image description here

我使用 opencv cv2.boundingRect()蒙版图像上绘制边界框。我创建了一个包含 bounding_Boxes 和坐标的列表 -- 如何添加预测类,即每个框显示最高概率的类?

contours,hierarchy = cv2.findContours(edged.copy(),cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
bounding_Boxes = [cv2.boundingRect(contour) for contour in contours]

此外,预测的类别散布在我的蒙版图像中,如下图所示。每种颜色表示一个标签。在我绘制 bounding_Boxes 时,某些区域可能包含其他标签。创建这些框以便每个框都包含一个预测类的“更好”方法可能是什么?

enter image description here

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)