torch == 1.8.1 torchvision == 0.9.1 onnx == 1.11.0 numpy == 1.18.5 opencv-python == 4.4.0.46 Pillow == 8.3.2