torch==1.6
torchvision==0.7
pillow
numpy
