TF-slim保存问题
tensorflow吧
全部回复
仅看楼主
level 1
海水有毒0 楼主
我训练的时候是没有任何问题的,每次 18张图片的训练,用的是inception_resnet_v2,保存之后,我取出模型用在测试图片的时候,每次测试一张图片,不管测试多少次,结果都是一样,比如第一次测试 分类a概率为百分之10,分类b概率为百分之20 ,分类c概率为百分之70,。那么不管我换什么图片这个概率都不会变,只有我再次用18张测试图片一起试的时候才行,用的是slim库。代码就是github上面的源码。不带保存,我自己在训练的地方加的保存
2017年05月25日 04点05分 1
level 1
训练都和往常一样了,导出模型后,在测试阶段调用相同的网络,参数is_training一定要设置成False。
?
logits = model.resnet(X, 100, is_training=False)
否则,可能会出现这种情况:所有的单个图像分类,最后几乎全被归为同一类。这可能就是训练模式设置反了的问题
2017年07月24日 09点07分 2
level 1
实际上是你自己理解的错误,batch=1的时候最终得到的y是分类几个的一维向量,只是通过计算准确率的时候用到了tf.argmax(y, 1)这个函数,获取一个向量的最大值的坐标,相当于获取本次分类最终会在哪一个类里面
2018年11月27日 08点11分 3
[滑稽],并不是理解错误,我记得当时是我调用slim库的封装模型方法有误
2018年11月29日 07点11分
后来按照官方调用slim库的方式调用结果正常,我当时没看用例就直接调函数跑模型了
2018年11月29日 07点11分
@海水有毒0 遇到了同样的问题,请问具体是怎么解决的?
2019年04月16日 13点04分
@芙玲的幻想曲 我当时是调用slim库的方式不对,当时没仔细看官方示例,用了错误的方式调用的,导致结果出问题
2019年04月17日 19点04分
1