欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

U-Net运行报错merge6 = merge([drop4,up6], mode = 'concat'...) TypeError: 'module' object is not callable

程序员文章站 2024-01-10 16:30:58
...

运行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

源代码地址:https://github.com/FENGShuanglang/unet