有没有办法导出Yolo v7模型的mask的多边形坐标？

Question

我正在使用 Wong Kin Yiu 的 存储库在我注释的数据集上训练 Yolo v7 模型。目标是在手术过程中制作心脏面罩。我已经成功使用模型创建了心脏的掩模，但据我所知，我只能导出边界框坐标。我需要能够导出蒙版的多边形坐标。

我尝试查看源代码和掩码创建函数，以查找是否有需要导出的参数。我什么也找不到；仅边界框数据。

Answer 1

你有口罩。

剩下的就是 转 它变成一个 凸包。 这在很多地方都有解释，包括 这里。