人工智能
如何使用tensorflow,本博会将从训练模型讲起,再到模型保存,c++/C如何调用该模型。并逐步在Windows中搭建模型使用环境,将模型转换为实际使用的产品
happy hell
主要从事网络通讯方面研究,目前领导的项目有jabber IM、企业版IM、视频会议、视频监控、GPS车载、p2p技术、IPTV等等项目。
目前从事android平台整体系统架构,以及自主SDK研发,关注meego
展开
-
tensorflow之模型保存
1、ckpt 2、pd def freeze_graph(input_checkpoint,output_graph): ''' :param input_checkpoint: :param output_graph: PB模型保存路径 :return: ''' # 指定输出的节点名称,该节点名称必须是原模型中存在的节点 # 直接用最...原创 2019-12-02 17:26:18 · 285 阅读 · 0 评论 -
tensorflow如何将ckpt转成pd
研究人工智能很久了,一直没写个文章,今天将之前碰到的问题,分享给大家。 如何将tensorflow的ckpt模型转化为pb模型 网上找了很多中方法,都不行。。。希望有缘人能看到,并解决该问题。 1、在训练的时候,找到你输出的接口,如标红,并将它打印出来。 # 加上DropOut,防止过拟合 dropout_fc = tf.layers.dropout(fc, dropout_pla...原创 2019-12-02 17:17:27 · 814 阅读 · 0 评论