unet代码实践

最新推荐文章于 2024-09-27 16:00:11 发布

原创

最新推荐文章于 2024-09-27 16:00:11 发布 · 3.1k 阅读

17 ·

CC 4.0 BY-SA版权

文章标签：

#unet #深度学习 #python #unet原理

博主记录uNet代码问题供大家参考。一是网络输入处因通道格式不同（Channel_first和Channel_last）会报Tensor尺寸错误，可修改通道文件或代码；二是部分代码网络特征拼接与原文不符，需调整降维及拼接轴参数。

最近看了一下uNet的代码，发现了一些问题，再次记录一下，供大家参考和学习。

1)网上的代码有一些在网络输入位置为inputs=((C,w,h))，但是你在运行的过程中会报一些Tensor的尺寸存在问题的错误，这是因为别人的代码是使用了Channel_first，而你的后端的通道格式是Channel_last，所以会出现尺寸问题，这里请大家注意。怎么改，大家网上搜一下，无非就是，要么你把你的通道文件改成Channel_first的形式，要么你就把代码改了就可以了。
2）有的代码在网络的特征拼接位置和原始的论文不符。
下面是原始代码里的拼接部分。

conv4 = Conv2D(512, (3, 3), activation="relu", padding="same")(pool3)
conv4 = Conv2D(512, (3, 3), activation="relu", padding="same")(conv4)
pool4 = MaxPooling2D