Skip to content

Latest commit

 

History

History
27 lines (14 loc) · 717 Bytes

README.md

File metadata and controls

27 lines (14 loc) · 717 Bytes

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