numpy
torch
pillow
os
