Skip to content

GIoU for rotated bounding boxes, vectorized and differentiable

License

Notifications You must be signed in to change notification settings

jw9730/ori-giou

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GIoU for oriented bounding boxes

This code is a PyTorch-based, vectorized, and differentiable implementation of Generalized union over intersection (GIoU) loss for rotated bounding boxes.

The base code for axis-aligned box manipulation and optimal matching was taken from DETR by Facebook Research.

For use cases, see test.py.

Demo

Red boxes are optimized by gradient descent in respect to blue boxes.

  1. Single box

single

  1. Regular boxes on grid

grid_regular

  1. Irregular boxes on grid

grid_irregular

  1. Random boxes

randombox1

Releases

No releases published

Packages

No packages published

Languages