深度学习
文章平均质量分 95
dyclg
这个作者很懒,什么都没留下…
展开
-
windows下编译tensorflow2.5.0 c++库并调用
近期使用python+keras+unet网络训练了一个图像分割模型(.h5),因为最终需要在C++中使用,所以需要2步转换:模型转换(h5 —> pb):需要的朋友可以参考一下之前写的文章。模型转换编译tensorflow,生成C++库:本文重点说明的内容。说明:keras没有c++相关接口,所以不能直接在c++中调用由keras.save生成的h5模型,需要将其转换为pb模型,由tensorflow c++相关接口来调用。1、开始之前明确需要编译的tensorflow版本,这个和原创 2021-07-02 19:55:34 · 4804 阅读 · 36 评论 -
keras训练的h5模型转换为pb模型
最近使用keras训练了一个图像分割的模型(.h5),但最终需要在C++中调用该模型,由于keras没有C++接口,所以需要将.h5模型转换为.pb模型后通过tensorflow C++接口进行调用。由于本人之前接触深度学习较少,很多东西不是很懂,所以在转换过程中遇到了很多问题,在此记录,共同学习。1、转换之前需要注意的点本人在转换过程中发现tensorflow1.x和2.x存在区别,所以在转换之前最好确定训练模型时使用的tensorflow版本,转换过程使用的环境尽量和训练模型使用的环境保持一致,原创 2021-06-28 18:22:55 · 5158 阅读 · 9 评论