一、遇到问题
使用eval输出为MIOU值的过程中,遇到了如下问题:
INFO:tensorflow:Waiting for new checkpoint at D:/zyt/DLProject/deeplab/datasets/Pascal_Voc/train_on_train_set/train
其中 D: 及其后面的是路径。
二、解决方案
不同的人遇到的情况不同,解决方案也不同。
1、训练过程未停止
有人遇到的问题是由于训练过程未停止,因为验证和训练交替进行,需要不断监听保存模型的网络,看是否有新的模型生成,只要有,就开始验证新的模型。等待训练停止以后会继续执行。
2、代码本身的问题
我运行eval.py时,训练已经结束了,自然不会出现上面的情况,这就要考虑一下代码本身了。
找到如下代码:
tf.contrib.training.evaluate_repeatedly(
master=FLAGS.master,
checkpoint_dir=FLAGS.checkpoint_dir,
eval_ops=[update_op],
max_number_of_evaluations=num_eval_iters,
hooks=hooks,
eval_interval_secs=FLAGS.eval_interval_secs)
修改为下面的内容:
tf.contrib.training.evaluate_repeatedly(
master=FLAGS.master,
checkpoint_dir=FLAGS.checkpoint_dir
)