torch
numpy
Pillow