numpy
matplotlib
torch
torchvision
scipy>=1.10.0
PyYAML
escnn
h5py
