pytorch-lightning
pandas
dill
opencv-python
matplotlib
albumentations
torchvision
torchmetrics
