Export PyTorch Model to ONNX – Convert a Custom Detection Model to ONNX

In this article, we train a custom PyTorch RetinaNet model and export it to ONNX format. Further, we run inference using it on the CUDA device.

The post Export PyTorch Model to ONNX – Convert a Custom Detection Model to ONNX appeared first on DebuggerCafe.

article cuda deep learning detection export format inference object-detection onnx onnx-runtime pytorch pytorch retinanet retinanet