matplotlib
numpy
opencv-python-headless
Pillow
PyYAML
scipy
torch
torchvision
tqdm
seaborn
pandas
gdown