【deeplab-eval 填坑】 运行eval.py时停留在Waiting for new checkpoint at...

一、遇到问题

使用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
    )

 

发布了244 篇原创文章 · 获赞 501 · 访问量 50万+

猜你喜欢

转载自blog.csdn.net/shuiyixin/article/details/90572658