pytorch-lightning>=1.7.0
torch>=1.12.0
pandas
dill
opencv-python
matplotlib
albumentations
torchvision>=0.13.0
torchmetrics>=0.9.0
