lilanxiao / Rotated_IoU

Differentiable IoU of rotated bounding boxes using Pytorch
MIT License
415 stars 64 forks source link

The computational cost of accurate rotated iou? #34

Closed colian closed 3 years ago

colian commented 3 years ago

Thanks for your work. I notice the computational cost of accurate rotated iou is very high. This code calculates and restores all the situations of intersection, so the GPU memory cost is also very high?

lilanxiao commented 3 years ago

yes, my code uses the brute force method. I do so because it's easier to make the brute force method parallelable and differentiable.

colian commented 3 years ago

Thanks for your reply, that really helps a lot.