Skip to content

More object detection models and backbones

Compare
Choose a tag to compare
@akihironitta akihironitta released this 20 Dec 17:15
· 194 commits to master since this release
ba899e8

[0.5.0] - 2021-12-20

Added

  • Added YOLO model (#552)
  • Added SRGAN, SRImageLoggerCallback, TVTDataModule, SRCelebA, SRMNIST, SRSTL10 (#466)
  • Added nn.Module support for FasterRCNN backbone (#661)
  • Added RetinaNet with torchvision backbones (#529)
  • Added Python 3.9 support (#786)

Changed

  • VAE now uses deterministic KL divergence during training, previously estimated KL divergence by random sampling (#760)

Removed

  • Removed PyTorch 1.6 support (#786)
  • Removed Python 3.6 support (#785)

Fixed

  • Fixed doctest fails with ImportError: cannot import name 'Env' from 'gym' (#751)
  • Fixed MoCo v2 missing Cosine Annealing learning scheduler (#757)

Contributors

@abhayraw1 @akihironitta @chris-clem @hoangtnm @nmichlo @oke-aditya @Programmer-RD-AI @senarvi

If we forgot someone due to not matching commit email with GitHub account, let us know :]