如何使用AI模型对多边形物体进行目标检测并将线画在检测到的物体上？

Question

如下图所示，这里有一排排的植物。我想检测每排植物并在每排植物上画出一条线。

当我输入新的图片或视频时都能像这样将线标注在每排植物上。

我试过使用opencv的线段边缘检测，但是这种植物无法检测到线段的边缘。
我也试过使用预训练的maskrcnn模型，但是输出的效果不是很理想，因为这个模型会将一个矩形框一起输出出来，而不仅是一条线。如下图所示。

请问用什么模型比较合适？以及该如何标注数据? 是在每张图片的每排植物上用线段标注还是用多边形将每排植物给框起来？单纯的用CNN模型吗？还是关键点检测啥的？

Answer 1

我可能会这样做标签：
以你图中的某一列（列c）植物为例：
标记列c最下端 较完整的植物的左右两个点A、B；
标记列c最上端 较完整的植物的左右两个点C、D；
即相当于每列植物有4个特征点，对所有列都如此操作做好数据集标签进行训练得到模型。
########
对于模型使用时预测的图片img，理想情况下检测出多对{A、B、C、D}；
这时需要opencv去处理下，使得每一对A、B找到自己的C、D，而不是另一列的C、D，否则错乱；
最后每对AB的中点与自己CD中点进行连线；