检测 YOLO 检测到的物体的角度

问题描述

我使用 YOLO 正确检测到上述注射器。现在我想知道如何测量注射器给出的大致角度?有什么想法吗?我需要正确预测注射器和表面之间的角度“有误差”。我想检测基线之间的偏差角,基线是注射器的图像下边缘。我不确定我的假设是否正确。另外,有没有办法根据注射器的方向在注射器中间画一条线?

Syringe Training Image

解决方法

如果您有用角度注释的数据,除了 bbox 之外,您还可以训练网络来预测角度。 如果你这样做,我会建议预测 2 个值,它们是角度的余弦和正弦值,而不是直接预测角度以避免在 -pi/pi 处环绕。