加载模型时发生ValueError :(“无法识别的关键字参数:”,dict_keys(['input_shape']))

时间:2020-04-23 13:20:41

标签: 3d python-3.6 tensorflow2.0 image-segmentation unity3d-unet

enter code here我在使用免费的gpu的地方使用google colab,需要tensorflow 2.0。输入层包含4个元组通道,是多个3d图像。我称之为一个函数:

    input_shape= input_shape=(4,96,96,96)
    model = build_model(input_shape=input_shape, output_channels=3)


   def build_model(input_shape=(4, 96,96,96), output_channels=3, weight_L2=0.1, weight_KL=0.1):
    ## Input Layer
   inp = tf.keras.Input(shape=[4,96,96,96],name='inp')
    ## The Initial Block
   x =  tf.keras.layers.Conv3D(
   filters=32,
   kernel_size=(3, 3, 3),
   strides=1,
   padding='same',
   data_format='channels_first',
   name='Input_x1')(inp)

   ## Dropout (0.2)
   x =  tf.keras.layers.Dropout(0.2)(x)

   ## Green Block x1 (output filters = 32)
   x1 = tf.green_block(x, tf.constant(32), name='x1')
   x =  tf.keras.layers.Conv3D(
   filters=32,
   kernel_size=(3, 3, 3),
   strides=2,
   padding='same',
   data_format='channels_first',
   name='Enc_DownSample_32')(x1)

我收到以下错误: ValueError跟踪(最近一次通话最后一次)

in()

---> 89模型= build_model(input_shape = input_shape,output_channels = 3) 90 91 model.fit(数据,标签,batch_size = 1,时代= 100)

1帧

输入中的

/usr/local/lib/python3.6/dist-packages/tensorflow/python/keras/engine/input_layer.py(形状,batch_size,名称,dtype,稀疏,张量,参差不齐,**变形)

286   if kwargs:

-> 287引发ValueError('无法识别的关键字参数:',kwargs.keys()) 288 289,如果shape为None且张量为None:

ValueError :(“无法识别的关键字参数:”,dict_keys(['input_shape']))

0 个答案:

没有答案