[question] loading ONNX models

leovandriel / caffe2_cpp_tutorial

C++ transcripts of the Caffe2 Python tutorials and other C++ example code

BSD 2-Clause "Simplified" License

431 stars 94 forks source link

[question] loading ONNX models #70

Open alexge233 opened 6 years ago

alexge233 commented 6 years ago

Hi Leonard,

Did you by any chance play with the new ONNX models, using C++? Seeing how caffe2 has moved into PyTorch and how a lot of top dogs are pushing towards ONNX I think it would be very useful. I have googled around but didn't find anything in C++, only a few python examples. Not even sure if onnx-caffe2 is built in (seems to have been merged into caffe2) or not.

Best regards, Alex

leovandriel commented 6 years ago

Hi Alex,

It look like onnx is now part of caffe2. I agree, it would be a good addition to the project. Currently I'm not actively developing in caffe2, but I'm happy to merge support for onnx. There seems to be a solid c++ base implementation in the main onnx repo.

alexge233 commented 6 years ago

Ok, I'll have a look at their code and see if there is an example to be written. I'm playing with pytorch atm but I'm still interested in the c++ side of it.