requests
tqdm
seaborn
scikit-learn
thop
imageio
argparse
torch
einops
pyyaml
tensorboard
psutil
pytorch_wavelets
opencv-python
torchvision
