Skip to content

0.2

Pre-release
Pre-release
Compare
Choose a tag to compare
@hgaiser hgaiser released this 03 Mar 09:53
05bd676

Changes since last release

  • Corrected FPN architecture as per paper.
  • Set default image size to minimum of 800px.
  • Change NMS to perform per-class NMS.
  • Small correction for bbox transform.
  • Add OID data generator.
  • Change default NMS threshold to 0.5.
  • Add MobileNet backbone.
  • Add tensorboard callback.
  • Add tool for debugging datasets.
  • Improve speed of data augmentation methods.
  • Add ability to resume training.
  • Add evaluation tool for custom datasets (only computes mAP at the moment).
  • Add skip_mismatch to weights loading, allows transfer learning from pretrained COCO model.

Credits to:
@awilliamson
@hgaiser
@de-vri-es
@mxvs
@wassname
@mkocabas
@lvaleriu