ValueError на Lambda

Question

Я создаю модель сегментации. Я добавляю следующие выводы:

 conv38_1 = Conv2D(filters=NUM_CLASSES, kernel_size=3, dilation_rate=(6,6), padding='same', name='fc1_voc12_c0')(relu38_1)
    conv38_2 = Conv2D(filters=NUM_CLASSES, kernel_size=3, dilation_rate=(12,12), padding='same', name='fc1_voc12_c1')(relu38_1)
    conv38_3 = Conv2D(filters=NUM_CLASSES, kernel_size=3, dilation_rate=(18,18), padding='same', name='fc1_voc12_c2')(relu38_1)
    conv38_4 = Conv2D(filters=NUM_CLASSES, kernel_size=3, dilation_rate=(24,24), padding='same', name='fc1_voc12_c3')(relu38_1)

    output   = Add(name='fc1_voc12')([conv38_1, conv38_2, conv38_3, conv38_4])
    output   = Lambda(lambda image: tf.image.resize_images(image, (H,W)))(output)

Однако я получаю странную ошибку следующим образом:

ValueError: Ошибка при проверке цели: ожидаемый lambda_1 будет иметь форму (321, 321, 1), но получилмассив с формой (321, 321, 2)

почему третий элемент формы имеет 2?есть ли способы разобраться?

Answer 1

Это (321,321,2) форма вашего y_train.

Это означает, что вы пытаетесь предсказать два класса / канала для каждого изображения.Ваша модель должна также вывести 2 класса.

NUM_CLASSES должно быть 2.