torch>=1.8.0
torchvision>=0.9.0
matplotlib
opencv-python
numpy
tqdm
pyyaml
pillow
scipy
wandb
sklearn
