torch
torchvision
torchaudio
timm
einops
thop
pytorch-fid
lpips
opencv-python
matplotlib
scikit-image
