tqdm
torch
wandb               #optional
torchvision
torchsummary
Pillow==7.2.2
opencv-python
