如何为新图像训练VGG-16网络?

时间:2018-06-24 12:17:35

标签: python-2.7 tensorflow image-processing jupyter-notebook conv-neural-network

我正在尝试为turorial中可用的代码提供新的输入和注释图像:http://warmspringwinds.github.io/tensorflow/tf-slim/2016/12/18/image-segmentation-with-tensorflow-using-cnns-and-conditional-random-fields/。我的输入图像为灰度(250 x 180 px),我将注解(250 x 180 px)归一化为[0 1]。但是,我无法使我的代码适用于新图像。我收到以下错误:

182损失,summary_string = sess.run([cross_entropy_sum,merged_summary_op],

-> 183 feed_dict = feed_dict_to_use)

InvalidArgumentError:logits和标签必须是可广播的:logits_size = [21504,2] labels_size = [29500,2] [[节点:softmax_cross_entropy_with_logits_sg = SoftmaxCrossEntropyWithLogits [T = DT_FLOAT,_device =“ / job:localhost / replica:0 / task:0 / device:CPU:0”](softmax_cross_entropy_with_logits_sg / Reshape,softmax_cross_entropy_strong>]

有人可以帮我吗?

0 个答案:

没有答案