torch
torchvision
matplotlib
opencv-python
numpy
tqdm
pyyaml
pillow
scipy
