pytorch-lightning==1.1.4 test-tube==0.7.1 numpy==1.17.2 tqdm>=4.36.1 pretrainedmodels==0.7.4 Pillow==6.2.0 fire==0.2.1 tensorboardX==1.9 streamlit==0.53.0 albumentations==0.4.6 imgaug==0.4.0 pytorch-ignite scikit-learn==0.23.2 seaborn==0.10.1 segmentation-models-pytorch torch torchvision