torch
numpy
matplotlib
scikit-learn
lightning
scikit-image
omegaconf
opencv-python
wandb
