einops
numpy
matplotlib
opencv-python
h5py
torch==1.7.0
torchvision==0.8.1