Update YOLO code to use onnx==1.9.0 (instead of 1.4.1)

jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet

https://jkjung-avt.github.io/

MIT License

1.75k stars 547 forks source link

Update YOLO code to use onnx==1.9.0 (instead of 1.4.1) #550

Closed jkjung-avt closed 2 years ago

jkjung-avt commented 2 years ago

The code has been tested with TensorRT 8 (onnx==1.9.0, aka opset 14, is OK). For TensorRT 7, users might need to fall back to onnx==1.6.0, aka opset 11.