einops==0.4.1 kornia==0.2.0 numpy==1.21.5 opencv_python==4.5.3.56 scikit_image==0.19.2 scikit_learn==1.1.3 scipy==1.7.3 tensorboardX==2.5.1 timm==0.4.12 torch==1.8.1+cu111