Skip to content

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.


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 albumentations and then update the augmentation pipeline as you see fit in the Albumentations class in utils/ An example is available in the YOLOv5 repository.

Other frameworks and libraries

Other you can see find at GitHub