小编wil*_*llz的帖子

如何使用 1 通道图像作为 VGG 模型的输入

我首先使用 3 通道图像作为 VGG16 模型的输入,没有任何问题:

input_images = Input(shape=(img_width, img_height, 3), name='image_input')
vgg_out = base_model(input_images)  # Here base_model is a VGG16
Run Code Online (Sandbox Code Playgroud)

现在我想改用 1 通道图像。所以我这样做了:

input_images = Input(shape=(img_width, img_height, 1), name='image_input')
repeat_2 = concatenate([input_images, input_images])
repeat_3 = concatenate([repeat_2, input_images])
vgg_out = base_model(repeat_3)  
Run Code Online (Sandbox Code Playgroud)

但我收到一条错误消息:

File "test.py", line 423, in <module>
model = Model(inputs=[input_images], outputs=[vgg_out])
File "C:\Users\wzhou\AppData\Local\Continuum\Anaconda2\envs\tensorflow\lib\site-packages\keras\legacy\interfaces.py", line 91, in wrapper
return func(*args, **kwargs)
File "C:\Users\wzhou\AppData\Local\Continuum\Anaconda2\envs\tensorflow\lib\site-packages\keras\engine\network.py", line 93, in __init__
self._init_graph_network(*args, **kwargs)
File "C:\Users\wzhou\AppData\Local\Continuum\Anaconda2\envs\tensorflow\lib\site-packages\keras\engine\network.py", line 237, in _init_graph_network
self.inputs, self.outputs)
File "C:\Users\wzhou\AppData\Local\Continuum\Anaconda2\envs\tensorflow\lib\site-packages\keras\engine\network.py", line 1430, …
Run Code Online (Sandbox Code Playgroud)

image keras vgg-net

4
推荐指数
1
解决办法
9677
查看次数

标签 统计

image ×1

keras ×1

vgg-net ×1