torch==1.10.0
torchvision==0.11.0
einops
timm
imageio
scikit-image
numpy
opencv-python