tensorflow
Gemma0906
I don't know where I am going, but I'm on the way...
展开
-
CNN多输出回归问题
最近用CNN做回归一直收敛不了,今天终于解决了,在此记录一下。 我的输出是一张图片,输出是70个系数值,原来不收敛是因为只对输入做了预处理,输出用的原始数据,网络最后的输出层也没加激活函数,label范围不固定。把输入输出都归一化到-1到1,最后一层用tanh激活就可以收敛了。用的VGG收敛速度还是有点慢需要再优化下网络。 ...原创 2020-08-27 15:05:02 · 4161 阅读 · 6 评论 -
Tensorflow导出pb模型,并在python和matlab下分别进行预测
tensorflow下训练完模型测试程序比较杂乱,特此整理一下。 1、我都是在linux下训练,windows下测试分析,训练保存模型如下所示。 2、然后调用frozen_model.py将模型进行固化原创 2020-07-16 10:10:39 · 980 阅读 · 0 评论