numpy
pandas
torch
PyYAML
albumentations
tqdm
matplotlib
opencv-python
scikit-learn
