Skip to content

OrientedDet v0.1.0

Choose a tag to compare

@jeffaudi jeffaudi released this 21 Jun 09:15

OrientedDet v0.1.0

First public release of OrientedDet.

Install from PyPI:

pip install oriented-det

Highlights

  • Core geometry (Polygon, QBox, RBox) and transforms
  • Rotated IoU, NMS, and optional GPU kernels
  • DOTA loader, tiling, augmentations, and oriented mAP evaluation
  • Airbus Playground CSV dataset support
  • Three baseline detectors:
    • Oriented R-CNN
    • Rotated Faster R-CNN
    • Rotated RetinaNet
  • JSON config training via odet train
  • odet CLI for training, inference, metrics, dataset tools, and pretrained downloads
  • Pretrained weights on Hugging Face Hub: dl4eo/oriented-det-pretrained

Model Zoo

DOTA le90 pretrained checkpoints include:

Model Slug eval-val mAP50
Oriented R-CNN 1× oriented_rcnn_dota_le90_1x 74.79%
Rotated Faster R-CNN 3× rotated_faster_rcnn_dota_le90_3x 76.41%
Rotated RetinaNet 3× rotated_retinanet_dota_le90_3x 71.52%
Rotated RetinaNet 1× rotated_retinanet_dota_le90_1x 64.14%

Download weights:

odet pretrained list
odet pretrained download oriented_rcnn_dota_le90_1x

Links