torch 
torchvision 
timm 
scipy 
pytorch-msssim 
Pillow