torch
torchvision
albumentations
opencv-python-headless
scikit-learn
Pillow
tqdm
numpy<2
