0%

PyTorch + Onnx + Caffe2 C++ Library

現在在科技業沒講個深度學習會被翻白眼,Prototype是一回事,放進Product又是另一回事
之前一般來說通用信的選擇是Tensorflow,雖說Prototype跟Product可以一起完成
不過C++那端難寫就算了,Python那邊也麻煩的要死,沒太多精力搞這個
後來看到PyTorchOnnx,以及Caffe2改變了想法
用易學易用的PyTorch建構出Onnx Mdoel,透過Onnx轉換成Caffe2 Model,加上對終端最佳化的Caffe2 Library
變成另外一種可行的解法
我自己的實驗結果就放在 GitHub
其中C++ Demo的部分是從Caffe2 Android Example那邊學來的
雖然看起來不多,不過真正讓他可以動倒是花了不少時間