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
cv2==4.6.0
