Skip to content

Object detection, segmentation and new layers
Compare
Choose a tag to compare
@frgfm frgfm released this 21 Jul 21:39
v0.1.2
59c3124
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

This minor release introduces new model tasks and training scripts.
In the release attachments, you will find remapped ReXNet ImageNet pretrained weights from https://github.com/clovaai/rexnet, ImageNette pretrained weights from the repo owner.

Note: holocron 0.1.2 requires PyTorch 1.5.1 and torchvision 0.6.1 or newer.

Highlights

models

Implementations of deep learning models
New

  • Added implementations of UNet (#43), UNet++ (#46), and UNet3+ (#47)
  • Added implementation of ResNet (#55), ReXNet (#56, #58, #59, #60)

Improvements

  • Updated Darknet pretrained models (#32)
  • Improved Darknet flexibility (#45)

Fixes

  • Fixed YOLO inference and loss (#38)

nn

Neural networks building blocks
New

  • Added implementations for Add2d (#35), NormConv (#34), SlimConv (#36, #49)
  • Added Dropblock implementation (#53)
  • Added implementations of SiLU/Swish (#54, #57)

Improvements

  • Improved efficiency of ConcatDownsample2d (#48)

optim

Optimizer and learning rate schedulers
New

  • Added implementation of TAdam (#52)

Improvements

  • Added support for rendering in notebooks (#39)
  • Fixed inplace add operator usage in optimizers (#40, #42)

Documentation

Online resources for potential users
Improvements

  • Improved docstring for better understanding (#37,

References

Verifications of the package well-being before release
New

  • Added training script for object detection (#41)
  • Added training script for semantic segmentation (#50)

Others

Improvements

Fixes

  • Fixed conda upload job (#33)
Assets 24
Loading