torch==1.3.0
torchvision==0.5.0
tqdm 
opencv-python 
scikit-learn 
matplotlib
h5py 
pyyaml
seaborn 
imageio 
easydict
tensorboardX
