Covert original YOLO model from Pytorch to Onnx, and do inference using backend Caffe2 or Tensorflow.
-
Updated
Jun 15, 2018 - Python
Covert original YOLO model from Pytorch to Onnx, and do inference using backend Caffe2 or Tensorflow.
This repository shows an example of how to use the ONNX standard to interoperate between different frameworks. In this example, we train a model with PyTorch and make predictions with Tensorflow, ONNX Runtime, and Caffe2.
Correctness Verification Performance Comparison of of different Deep Learning Frameworks such as Pytorch, Caffe2 and Tensorflow using ONNX format models.
Add a description, image, and links to the onnx-caffe2 topic page so that developers can more easily learn about it.
To associate your repository with the onnx-caffe2 topic, visit your repo's landing page and select "manage topics."