torch
torchvision
einops
numpy
pillow
tifffile
