如何将二值分割掩码转为YOLO Segmentation文本文件？问题与处理

最初的实现流程 #

我原本尝试通过以下步骤将二值分割掩码转换为YOLO Segmentation格式的文本标注文件：

• 将彩色掩码转换为仅含单类别的二值化掩码
• 对二值掩码执行轮廓检测，提取掩码的边缘点位
• 展平检测得到的轮廓点数组
• 基于角度大幅变化的规则保留关键点位，试图精简过多的轮廓点
• 按照YOLO Segmentation的标注格式（cls_name x1 y1 x2 y2 .... xn yn）保存处理后的点位数据

遇到的核心问题 #

实际落地时发现两个无法解决的问题：

1. 轮廓点精简的效果不理想，没能找到合适的关键点筛选策略
2. 轮廓检测会误识别图像的外边框，导致重叠的标签被错误合并成包含大量无关区域的大区块，标注结果完全不符合需求

最终解决方案 #

放弃了基于轮廓检测的转格式方案，改用UNET模型处理掩码数据，最终得到了符合预期的理想输出。

内容的提问来源于stack exchange，提问作者AHFromHogwarts