运行U-net进行图像分割时报错:
merge6 = merge([drop4,up6], mode = 'concat', concat_axis = 3)
TypeError: 'module' object is not callable
原因
tensor版本更新后,语法规则发生了改变,代码也要做出相应调整
解决办法:
把unet.py下的代码更改一下
merge6 = merge([crop4,up6], mode = 'concat', concat_axis = 3)
改成:
merge6 = concatenate([drop4,up6],axis=3)
merge7,8,9同理,按照上述规则更改 注意后面括号里的数字要与前面对应
转载自:https://blog.csdn.net/DoReAGON/article/details/83314022
unet原文及源代码链接:
原文:https://blog.csdn.net/mieleizhi0522/article/details/82217677