albumentations==1.4.18
numpy
opencv-python-headless
PyYAML
tqdm
matplotlib
pytest>=7.0.0
black>=22.0.0
isort>=5.0.0
flake8>=4.0.0

[cuda]
torch>=2.4.0
torchvision>=0.19.0
