【发布时间】:2018-09-15 10:09:54
【问题描述】:
我正在尝试使用 deeplab3+ tensorflow 实现来微调新数据集(具有不同数量的类)。我将数据集转换为 tfrecords 并开始 train 模型没有问题。现在我想评估新的检查点,运行evaluation 脚本,我得到一个形状不匹配错误。
tensorflow.python.framework.errors_impl.InvalidArgumentError:形状不匹配
模型实现使用tensorflow.slim,我认为我的问题与功能有关
slim.evaluation.evaluation_loop(
master=FLAGS.master,
checkpoint_dir=FLAGS.checkpoint_dir,
logdir=FLAGS.eval_logdir,
num_evals=num_batches,
eval_op=list(metrics_to_updates.values()),
max_number_of_evaluations=num_eval_iters,
eval_interval_secs=FLAGS.eval_interval_secs,
hooks=[tf_debug.LocalCLIDebugHook()]))
我的错误日志
`tensorflow.python.framework.errors_impl.InvalidArgumentError: Shape mismatch in tuple component 1. Expected [513,513,3], got [2448,2448,3]
[[Node: batch/padding_fifo_queue_enqueue = QueueEnqueueV2[Tcomponents=[DT_INT64, DT_FLOAT, DT_STRING, DT_INT32, DT_UINT8, DT_INT64], timeout_ms=-1, _device="/job:localhost/replica:0/task:0/device:CPU:0"](batch/padding_fifo_queue, Reshape_3/_4659, add_2/_4661, Reshape_1, add_3/_4663, case/cond/Merge/_4665, Reshape_6/_4667)]]`
我不明白这个错误,因为该实现在训练和验证期间使用了相同的 preprocessing 例程。我也尝试使用 tf_debug.LocalCLIDebugHook() 进行调试,但没有成功。
【问题讨论】:
标签: python tensorflow deep-learning