Логиты и метки должны быть транслируемыми: logits_size = [82944,2] label_size = [90000,2]

Question

Я работаю над проектом семантической сегментации кровеносных сосудов сетчатки с Tensorflow с моделью MobileUNet, и я получил эту ошибку:

    InvalidArgumentError (see above for traceback): logits and labels must 
    be broadcastable: logits_size=[82944,2] labels_size=[90000,2] 
[[Node: 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_with_logits_sg/Reshape_1)]]

Здесь мой код выглядит следующим образом:

network=network = build_mobile_unet(net_input, preset_model = args.model, num_classes=num_classes) 
net_input = tf.placeholder(tf.float32,shape=[None,None,None,3]) 
net_output = tf.placeholder(tf.float32,shape=[None,None,None,num_classes])
losses = tf.nn.softmax_cross_entropy_with_logits(logits=network, labels=net_output) 
cost = tf.reduce_mean(losses)

opt = tf.train.AdamOptimizer(0.001).minimize(cost)

init = tf.initialize_all_variables () _, current = sess.run ([opt, cost], feed_dict = {net_input: input_image_batch, net_output: segmented_image_batch})

Исходное изображение - 300x300, и оно находится в цветовом пространстве RGB. Вывод представляет собой двоичное изображение того же размера, что и ввод.

Может ли кто-нибудь мне помочь?

Answer 1

Мы ответили на эту проблему, которая также связана с архитектурой. Поймите это по следующей ссылке Входными данными для изменения формы является тензор с 37632 значениями, но запрошенная форма имеет 150528 Дайте нам знать, если у вас возникнут проблемы