需要做一個實際產品,主要開發環境為VS,C++,需要一個性能過得去(最好是自動GPU/CPU雙支持)的庫實現簡單的CNN網路,因為可能用於商業授權協議上避免GPL。求推薦,謝謝!


2020年了,目前感覺比較好用直觀簡單的是PyTorch的C++ api,stacktrace報錯清晰性能也不錯,api運行設計也很簡單。


用pytorch訓練好模型後轉換成onnx格式的模型文件,再用onnx runtime c++ 版載入模型就可以了,這也是一種方式,而且比較通用的是大多數框架的模型都可以轉換成onnx格式的,pytorch,tensorflow,mxnet等等


mxnet可以,vc++支持很好。另外Dlib或者樓上所說的darknet也不錯,不過darknet對windows並不友好,要改代碼。


我的最愛darknet,純C和CUDA編程,不依賴第三方資源。注意一下作者提供的庫有內存資源泄露,修復一下就好。

目前這方面開源做的好的有騰訊的ncnn,阿里的MNN等,在PC端、移動端均可以使用。當然了早期的caffe,mxnet都是可以的。不過就使用而言我個人覺得騰訊和阿里開源的框架最方便也比較好用。


可以試試用OpenCV調用tensorflow或者pytorch訓練的模型


同意樓上,2021了,必須得上pytorch c++ api了,完美契合pytorch,C++可訓練,可載入預訓練權重,增量訓練。不要臉分享一個自己實現的分割項目。後續可能會有DetectionCPP...歡迎關注,沒錯,我就是來推廣的haha


推薦閱讀:
相关文章