首先我还是要喷一下坑了我好久的垃圾文章,在大多数与我标题相关的文章当中,99%的文章只写了一句话,在程序的开头加上tf.reset_default_graph(),而且连具体位置都没写,查过stackoverflow之后,才知道应该加载import tensorflow as tf。
但是问题来了,应该有不少同学跟我一样的情况,加了之后并没有作用,那么进入正题,在我尝试多次之后,终于发现并解决了我程序里的bug,这里写一下,给大家做一个思路上的参考。
我使用slim自己复现了一个cnn模型,在我的训练模型得文件当中,我定义了两个操作,第一个是训练集的训练操作,一个是验证集的训练操作,代码如下(以下是bug解决之后的代码,先放在这里)以下两行是正确的的代码!!!!!
with tf.variable_scope("VGG_19"):
pre,soft_max,predictions=VGG_19(inputs=train_images,is_training=True,NUM_CLASSES=NUM_CLASSES)
tf.get_variable_scope().reuse_variables()
_,_,val_pre=VGG_19(inputs=val_images,is_training=False,NUM_CLASSES=NUM_CLASSES)
这里关注reuse参数