需要做一个实际产品,主要开发环境为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


推荐阅读:
相关文章