输出6个预测值如何转为poly的顶点坐标8个值结果？

jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.

Apache License 2.0

537 stars 113 forks source link

Closed MMountains closed 3 years ago

MMountains commented 3 years ago

尊敬的作者您好，感谢您的代码分享！在使用的过程中需要将输出结果转化为顶点的8个坐标值，请问输出的6个预测值分别是[center_x, center_y, w, h, theta, confidence]吗？还是说表示其他值？感谢😀

jbwang1997 commented 3 years ago

六个值的预测就是[center_x, center_y, w, h, theta, confidence]。

如果要转换为8个点的坐标可以先将confidence split下来。tensor使用mmdet.core.bbox2type(bboxes, 'poly'), np.ndarray使用BboxToolkit.bbox2type(bboxes, 'poly')来进行转换

MMountains commented 3 years ago

好的，感谢！