albumentations==1.3.1 numpy==1.26.0 opencv-python==4.8.1.78 torch==2.1.0 torchvision==0.16.0