- 博客(2)
- 收藏
- 关注
RSS订阅原创 tensorflow加载与保存模型
问题：1.训练好分类模型，比如训练保存了一个10分类的模型，但是实际用的时候呢，分类数可能会改变，但是还想继续使用前面保存的模型。那么相当于是只加载前几层的参数，最后一层做一些修改。2.加载预训练模型时，预训练模型缺少网络中定义的变量保存模型saver = tf.train.Saver()saver.save(sess,“model.ckpt”)加载模型saver.restore(sess,“model.ckpt”)不传参数时，相当于是保存了所有的参数，然后加载所有的参数。加载模型时变量缺
2021-01-10 16:14:37
252
原创 超详细的卷积后大小的计算公式
计算公式定义定义几个参数输入图片大小 W×W卷积核大小 F×F步长 Spadding的像素数 P于是我们可以得出计算公式为：N = (W − F + 2P )/S+1输出图片大小为 N×N以resnet50为例，输入为[1,3,224,224]，其中1为batchsize，3为通道数，224为height和width。经过第一层卷积后，其大小为[1,64,112,112]in_channels= 3//输入通道out_channels= 64 //输出通道nn.Conv2d(in_
2021-01-07 20:15:58
41319
10
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人