torch>=1.12.1
torchvision>=0.13.1
pytorch-lightning>=1.6.4
torchmetrics>=0.9.1
matplotlib>=3.1.3
numpy>=1.21.6
opencv-python
