torch == 1.5.0 torchvision == 0.6.0 onnx == 1.7.0 numpy == 1.21.1 Pillow == 8.2.0 opencv_python == 4.5.3.56