lilanxiao / Rotated_IoU

Differentiable IoU of rotated bounding boxes using Pytorch
MIT License
412 stars 62 forks source link

为什么input shape是 B,N,4,2? #52

Open zuoningz opened 11 months ago

zuoningz commented 11 months ago

Hello 大佬,

想问一下demo.py和oriented_iou_loss.py里为什么数据的shape是B,N,4,2? 为什么最后一个dimension是2? 不应该是B,N,4,1吗? 代表box每一个顶点的坐标

Screenshot 2023-10-16 at 10 41 21 AM Screenshot 2023-10-16 at 10 43 31 AM