Skip to content

Rotated detections

Pre-release
Pre-release
Compare
Choose a tag to compare
@james-nvidia james-nvidia released this 12 Mar 23:25
v0.2.0
2db6b07
This commit was signed with the committer’s verified signature.
yashv28 Yash Vardhan
GPG key ID: F27A0D0D506ED8B8
Learn about vigilant mode.

Version 0.2.0 introduces rotated detections.

Added

  • train arguments:
    • --rotated-bbox: Trains a model is predict rotated bounding boxes [x, y, w, h, theta] instead of axis aligned boxes [x, y, w, h].
  • infer arguments:
    • --rotated-bbox: Infer a rotated model.

Changed

The project has reverted to the name Object Detection Toolkit (ODTK), to better reflect the multi-network nature of the repo.

  • retinanet has been replaced with odtk. All subcommands remain the same.

Limitations

  • Models trained using the --rotated-bbox flag cannot be exported to ONNX or a TensorRT Engine.
  • PyTorch raises two warnings which can be ignored:

Warning 1: NCCL watchdog

[E ProcessGroupNCCL.cpp:284] NCCL watchdog thread terminated

Warning 2: Save state warning

/opt/conda/lib/python3.6/site-packages/torch/optim/lr_scheduler.py:201: UserWarning: Please also save or load the state of the optimzer when saving or loading the scheduler.
  warnings.warn(SAVE_STATE_WARNING, UserWarning)
Assets 2
Loading