torch
torchvision
gdown
opencv-python