numpy
matplotlib
torch
torchvision
h5py
tqdm
pillow
einops
opencv-python