Question

我正在使用Resnet50进行转学习。后端是张量流。我试图在Resnet上面再堆叠三层，但是失败并出现以下错误：

public function sendContactMail(ContactFormRequest $request)
    {
        \Mail::to('laurent@kirepo.lu')->send(new ContactForm($request));
    }

堆叠两个模型的代码如下：

Exception: The shape of the input to "Flatten" is not fully defined (got (None, None, 2048). 
Make sure to pass a complete "input_shape" or "batch_input_shape" argument to the first layer in your model.

Answer 1

包含include_top = False选项的最后一层resnet已经展平，您不需要另一个展平图层。

Answer 2

在实例化Resnet时，您必须显式输入input_shape。 model = ResNet50(include_top=False, weights='imagenet',input_shape=(224,224,3))

如果您将theano作为后端，则必须将频道数设置为第一个： model = ResNet50(include_top=False, weights='imagenet',input_shape=(3,224,224))

使用Resnet50进行Keras传输学习失败，出现异常

2 个答案: