Incorrect Albumentations version in requirements.txt (needs to be >1.0.3 but <2.0)

[ViKarp]

Search before asking

• [x] I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Training

Bug

When installing dependencies for classification with augmentation in YOLOv5 from the requirements.txt file, I found that the version of the albumentations library is pinned to >=1.0.3. However, proper augmentation functionality requires a version of albumentations higher or equal than 1.0.3, yet lower than 2.0.

Running the scripts with version >=2 can lead to errors.

Environment

Ubuntu 22.04 Python 3.10 Default requirements.txt from the repository

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

• [x] Yes I'd like to help by submitting a PR!

[UltralyticsAssistant]

👋 Hello @ViKarp, thank you for your interest in YOLOv5 🚀 and for reporting this possible issue with Albumentations!

For 🐛 Bug Reports, could you please provide a minimum reproducible example to help us debug the versioning issue? The details you’ve provided are helpful, but a short snippet or specific script demonstrating the issue with Albumentations' version would be greatly appreciated.

For reference, please ensure you're using the correct environment:

Requirements

Python>=3.8.0 with all dependencies installed from the requirements.txt. To set up the environment, follow these steps:

git clone https://github.com/ultralytics/yolov5  # clone
cd yolov5
pip install -r requirements.txt  # install requirements

If you encounter this issue while using one of our verified environments, we’d recommend double-checking your setup against the options below:

Verified Environments

• Notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Additionally, you might want to check out our ⭐️ guides to ensure everything is configured properly.

Status

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are currently passing. CI includes tests for training, validation, inference, export, and benchmarks.

This is an automated response to acknowledge your query; an Ultralytics engineer will assist you soon. Thank you for helping us improve YOLOv5! 🚀✨

[pderrenger]

@ViKarp thank you for catching this! We welcome your PR to update the Albumentations version constraint in the YOLOv5 requirements.txt to albumentations>=1.0.3,<2.0.0. The team will review it promptly. For implementation details, see the Albumentations integration documentation.

[github-actions[bot]]

👋 Hello there! We wanted to give you a friendly reminder that this issue has not had any recent activity and may be closed soon, but don't worry - you can always reopen it if needed. If you still have any questions or concerns, please feel free to let us know how we can help.

For additional resources and information, please see the links below:

• Docs: https://docs.ultralytics.com
• HUB: https://hub.ultralytics.com
• Community: https://community.ultralytics.com

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLO 🚀 and Vision AI ⭐