torch
torchvision
matplotlib
tqdm
wandb
opencv_python
numpy
Pillow
tqdm
