pytorh>=0.4.0
numpy
imageio
scipy
tensorboardX
matplotlib
scikit-image
h5py