You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

Unet图像分割训练时出现NoneType类型错误求助

Unet图像分割训练报错排查求助

我正在用Unet做图像分割,训练数据x_train形状为(557, 100, 100, 3),y_train形状为(557, 100, 100, 3),已确认数据无None值。模型各层形状已调整,模型summary如下:

Layer (type)                Output Shape                 Param #   Connected to
==================================================================================================
input_1 (InputLayer)        [(None, 100, 100, 3)]        0         []

conv2d (Conv2D)             (None, 100, 100, 32)         2432      ['input_1[0][0]']

dropout (Dropout)           (None, 100, 100, 32)         0         ['conv2d[0][0]']

max_pooling2d (MaxPooling2D)(None, 50, 50, 32)           0         ['dropout[0][0]']

conv2d_1 (Conv2D)           (None, 50, 50, 32)           25632     ['max_pooling2d[0][0]']

dropout_1 (Dropout)         (None, 50, 50, 32)           0         ['conv2d_1[0][0]']

max_pooling2d_1 (MaxPooling2D)(None, 25, 25, 32)         0         ['dropout_1[0][0]']

conv2d_2 (Conv2D)           (None, 25, 25, 32)           9248      ['max_pooling2d_1[0][0]']

dropout_2 (Dropout)         (None, 25, 25, 32)           0         ['conv2d_2[0][0]']

max_pooling2d_2 (MaxPooling2D)(None, 9, 9, 32)           0         ['dropout_2[0][0]']

conv2d_3 (Conv2D)           (None, 9, 9, 256)            73984     ['max_pooling2d_2[0][0]']

up_sampling2d (UpSampling2D)(None, 27, 27, 256)          0         ['conv2d_3[0][0]']

zero_padding2d (ZeroPadding2D)(None, 27, 27, 32)         0         ['conv2d_2[0][0]']

concatenate (Concatenate)   (None, 27, 27, 288)          0         ['up_sampling2d[0][0]',
                                                                     'zero_padding2d[0][0]']

conv2d_4 (Conv2D)           (None, 25, 25, 32)           82976     ['concatenate[0][0]']

dropout_3 (Dropout)         (None, 25, 25, 32)           0         ['conv2d_4[0][0]']

up_sampling2d_1 (UpSampling2D)(None, 50, 50, 32)         0         ['dropout_3[0][0]']

concatenate_1 (Concatenate) (None, 50, 50, 64)           0         ['up_sampling2d_1[0][0]',
                                                                     'conv2d_1[0][0]']

conv2d_5 (Conv2D)           (None, 46, 46, 32)           51232     ['concatenate_1[0][0]']

dropout_4 (Dropout)         (None, 46, 46, 32)           0         ['conv2d_5[0][0]']

up_sampling2d_2 (UpSampling2D)(None, 92, 92, 32)         0         ['dropout_4[0][0]']

zero_padding2d_1 (ZeroPadding2D)(None, 100, 100, 32)     0         ['up_sampling2d_2[0][0]']

concatenate_2 (Concatenate) (None, 100, 100, 64)         0         ['zero_padding2d_1[0][0]',
                                                                     'conv2d[0][0]']

conv2d_6 (Conv2D)           (None, 100, 100, 3)          1731      ['concatenate_2[0][0]']

==================================================================================================

我的实现代码如下:

input = keras.Input(shape=(100,100,3))
x1 = layers.Conv2D(32, 5, activation="relu", padding="same")(input)
x = layers.Dropout(0.1)(x1)
x = layers.MaxPooling2D((2,2), padding="same")(x)
x2 = layers.Conv2D(32, 5, activation="relu", padding="same")(x)
x = layers.Dropout(0.1)(x2)
x = layers.MaxPooling2D((2,2), padding="same")(x)
x3 = layers.Conv2D(32, 3, activation="relu", padding="same")(x)
x = layers.Dropout(0.1)(x3)
x = layers.MaxPooling2D((3,3), padding="same")(x)

x = layers.Conv2D(256, 3, activation="relu", padding="same")(x)

x = layers.UpSampling2D(size=(3,3))(x)
x3 = layers.ZeroPadding2D(padding=(1, 1))(x3)
x = layers.concatenate([x, x3], axis=-1)
x = layers.Conv2D(32, 3, activation="relu", padding="valid")(x)
x = layers.Dropout(0.1)(x)

x = layers.UpSampling2D(size=(2,2))(x)
#x2 = layers.ZeroPadding2D(2)(x2)
x = layers.concatenate([x, x2], axis=-1)
x = layers.Conv2D(32, 5, activation="relu", padding="valid")(x)
x = layers.Dropout(0.1)(x)

x = layers.UpSampling2D(size=(2,2))(x)
x = layers.ZeroPadding2D(padding=(4, 4))(x)
x = layers.concatenate([x, x1], axis=-1)
output = layers.Conv2D(3, 3, activation="relu", padding="same")(x)

model = keras.Model(inputs=input, outputs=output)
model.summary() #only for printing purposes
model.compile(
    optimizer='adam',
    loss='sparse_categorical_crossentropy',
    metrics=['accuracy']
    )
model.fit(x, y, epochs=100, batch_size=16)

训练时出现如下错误:

WARNING:tensorflow:Keras is training/fitting/evaluating on array-like data. Keras may not be optimized for this format, so if your input data format is supported by TensorFlow I/O we recommend using that to load a Dataset instead.
Traceback (most recent call last):
  File "D:\<mifolder>\ML-experiments\test\butterfly.py", line 189, in <module>
    model = simple_segmentation(x_train, y_train)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\<mifolder>\ML-experiments\test\butterfly.py", line 167, in simple_segmentation
    model.fit(x, y, epochs=100, batch_size=16)
  File "D:\<mifolder>\venv\Lib\site-packages\keras\src\utils\traceback_utils.py", line 70, in error_handler
    raise e.with_traceback(filtered_tb) from None
  File "D:\<mifolder>\venv\Lib\site-packages\keras\src\engine\data_adapter.py", line 266, in <genexpr>
    int(i.shape[0]) for i in tf.nest.flatten(inputs)
    ^^^^^^^^^^^^^^^
TypeError: int() argument must be a string, a bytes-like object or a real number, not 'NoneType'

已确认模型各层形状正常且无None值,但不清楚为何触发该错误,请求帮助排查原因并解决。

内容的提问来源于stack exchange,提问作者vincegeratorix

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.07 09:45:01