torch
pillow
numpy
