numpy
Pillow
PyYAML
scikit_learn
thop
torch
torchvision
tqdm
