albumentations
numpy
opencv-python-headless
psutil
pyyaml
scikit-learn
torch
torchvision
tqdm
pytest
twine
