用户登录
用户注册

分享至

tensorflow 预训练

  • 作者: 大湿胸mw
  • 来源: 51数据库
  • 2020-10-04
Tensorflow C++ API调用预训练模型和生产环境编译

试着写了一个python 中训练NN模型,用freeze_graph.py小工具固定参数到 .pb文件中,然后在C++中读取session->Run()来预测的完整demo。之后用用Bazel编译或者在自己的环境下一来libtensorflow_cc.so 编译。

http://blog.csdn.net/rockingdingo/article/details/75452711



  一般都是在py环境下训练,并且把weights和net都保存到一个protobuf的graph里,c++可以new个session,readbinaryProto,再session->create一下py当时的graph就恢复了,然后一样是session->run,feed数据,fetch拿到结果,tf使用bazel编译的,那堆工具链老复杂了,慢慢研究吧



  https://www.zhihu.com/question/41667903?from=profile_question_card



  厉害,都研究tensorflow了,有啥心得能否分享?
其实我只是知道tensorflow,完全没玩过
软件
前端设计
程序设计
Java相关