numpy
Pillow
matplotlib
torch
