torch>=1.6
torchvision>=0.7
pillow
numpy
