torch>=1.7
torchvision
numpy
Pillow
