einops
numpy
matplotlib
opencv-python
h5py
torch
torchvision