Frameworks and libraries that use Albumentations¶
MMDetection is an open source object detection toolbox based on PyTorch. It is a part of the OpenMMLab project.
- To install MMDetection with Albumentations follow the installation instructions.
- MMDetection has an example config with augmentations from Albumentations.
YOLOv5 🚀 is a family of object detection architectures and models pretrained on the COCO dataset, and represents Ultralytics open-source research into future vision AI methods, incorporating lessons learned and best practices evolved over thousands of hours of research and development.
- To use Albumentations along with YOLOv5 simply
pip install -U albumentationsand then update the augmentation pipeline as you see fit in the Albumentations class in
utils/augmentations.py. An example is available in the YOLOv5 repository.