问题描述:
跑U-Net模型代码(基于Keras)的时候,遇到了如下问题:
Traceback (most recent call last):
File "/home/user7/myproject/Codes/unet.py", line 107, in <module>
myunet.train()
File "/home/user7/myproject/Codes/unet.py", line 80, in train
model = self.get_unet()
File "/home/user7/myproject/Codes/unet.py", line 45, in get_unet
merge6 = merge([drop4, up6], mode='concat', concat_axis=3)
TypeError: 'module' object is not callable
查阅keras官方文档“Merge层”一节后,分析可能原因是当前版本的keras融合矩阵不再用merge
,而是用keras.layers.Concatenate
。
解决办法:
将出错代码
merge6 = merge([drop4, up6], mode='concat', concat_axis=3)
修改为:
merge6 = Concatenate(axis=3)([drop4, up6])