torch
Pillow
numpy 