torch
torchvision
pillow
numpy
scipy
einops
opencv-python-headless