pybind11
opencv-python
pillow
scipy
tqdm
torch
torchvision
pytorch-msssim
lightning
tensorboard
wandb>=0.12.10
