torch == 1.5.0
torchvision == 0.6.0
onnx == 1.7.0
numpy == 1.20.3
Pillow == 8.2.0
opencv-python == 4.5.2.54
albumentations == 0.5.2