ResUnet实现（tensorflow2）

最新推荐文章于 2024-03-18 10:11:30 发布
最新推荐文章于 2024-03-18 10:11:30 发布
阅读量1.6w 收藏 71
点赞数 4
分类专栏： tensorflow
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_37663944/article/details/103919125
版权

  我们将在本篇博客实现Resunet。
  首先上Unet图
在这里插入图片描述
然后结合Unet和Resnet，就是一个新的网络。

def bn_act(x, act=True):
    'batch normalization layer with an optinal activation layer'
    x = tf.keras.layers.BatchNormalization()(x)
    if act == True:
        x = tf.keras.layers.Activation('relu')(x)
    return x

以下方框代表数据，圆形代表方法，虚线代表判断
bn_act方法图
if
x
BN
relu
z 
def conv_block(x, filters, kernel_size=3, padding='same', strides=1):
    'convolutional layer which always uses the batch normalization layer'
    conv = bn_act(x)
    conv = Conv2D(filters, kernel_size, padding=padding, strides=strides)(conv)
    return conv

conv_block方法图
x
bn_act
conv2d
z

stem方法图

def stem(x, filters, kernel_size=3, padding='same', strides=1):
    conv = Conv2D(filters, kernel_size, padding=padding, strides=strides)(x)
    conv = conv_block(conv, filters, kernel_size, padding, strides)
    shortcut = Conv2D(filters, kernel_size=1, padding=padding, strides=strides)(x)
    shortcut = bn_act(shortcut, act=False)
    output = Add()([conv, shortcut])
    return output
x
conv2d
conv_block
+
conv2d
bn_act
output

residual_block方法图

 def residual_block(x, filters, kernel_size=3, padding='same', strides=1):
    res = conv_block(x, filters, k_size, padding, strides)
    res = conv_block(res, filters, k_size, padding, 1)
    shortcut = Conv2D(filters, kernel_size, padding=padding, strides=strides)(x)
    shortcut = bn_act(shortcut, act=False)
    output = Add()([shortcut, res])
    return output
x
conv_block
conv_block
conv2d
bn_act
+
output 
def upsample_concat_block(x, xskip):
    u = UpSampling2D((2,2))(x)
    c = Concatenate()([u, xskip])
    return c

ResUNet方法图

 def ResUNet(img_h, img_w):
    f = [16, 32, 64, 128, 256]
    inputs = Input((img_h, img_w, 1))
    
    ## Encoder
    e0 = inputs
    e1 = stem(e0, f[0])
    e2 = residual_block(e1, f[1], strides=2)
    e3 = residual_block(e2, f[2], strides=2)
    e4 = residual_block(e3, f[3], strides=2)
    e5 = residual_block(e4, f[4], strides=2)
    
    ## Bridge
    b0 = conv_block(e5, f[4], strides=1)
    b1 = conv_block(b0, f[4], strides=1)
    
    ## Decoder
    u1 = upsample_concat_block(b1, e4)
    d1 = residual_block(u1, f[4])
    
    u2 = upsample_concat_block(d1, e3)
    d2 = residual_block(u2, f[3])
    
    u3 = upsample_concat_block(d2, e2)
    d3 = residual_block(u3, f[2])
    
    u4 = upsample_concat_block(d3, e1)
    d4 = residual_block(u4, f[1])
    
    outputs = tf.keras.layers.Conv2D(4, (1, 1), padding="same", activation="sigmoid")(d4)
    model = tf.keras.models.Model(inputs, outputs)
    return model

  这个图就不画了。
可以看下这个模型的概括。

__________________________________________________________________________________________________
Layer (type)                    Output Shape         Param #     Connected to                     
==================================================================================================
input_1 (InputLayer)            [(None, 256, 800, 1) 0                                            
__________________________________________________________________________________________________
conv2d_167 (Conv2D)             (None, 256, 800, 16) 160         input_1[0][0]                    
__________________________________________________________________________________________________
batch_normalization_75 (BatchNo (None, 256, 800, 16) 64          conv2d_167[0][0]                 
__________________________________________________________________________________________________
activation_74 (Activation)      (None, 256, 800, 16) 0           batch_normalization_75[0][0]     
__________________________________________________________________________________________________
conv2d_169 (Conv2D)             (None, 256, 800, 16) 32          input_1[0][0]                    
__________________________________________________________________________________________________
conv2d_168 (Conv2D)             (None, 256, 800, 16) 2320        activation_74[0][0]              
__________________________________________________________________________________________________
batch_normalization_76 (BatchNo (None, 256, 800, 16) 64          conv2d_169[0][0]                 
__________________________________________________________________________________________________
add (Add)                       (None, 256, 800, 16) 0           conv2d_168[0][0]                 
                                                                 batch_normalization_76[0][0]     
__________________________________________________________________________________________________
batch_normalization_77 (BatchNo (None, 256, 800, 16) 64          add[0][0]                        
__________________________________________________________________________________________________
activation_75 (Activation)      (None, 256, 800, 16) 0           batch_normalization_77[0][0]     
__________________________________________________________________________________________________
conv2d_170 (Conv2D)             (None, 128, 400, 32) 4640        activation_75[0][0]              
__________________________________________________________________________________________________
batch_normalization_78 (BatchNo (None, 128, 400, 32) 128         conv2d_170[0][0]                 
__________________________________________________________________________________________________
conv2d_172 (Conv2D)             (None, 128, 400, 32) 4640        add[0][0]                        
__________________________________________________________________________________________________
activation_76 (Activation)      (None, 128, 400, 32) 0           batch_normalization_78[0][0]     
__________________________________________________________________________________________________
batch_normalization_79 (BatchNo (None, 128, 400, 32) 128         conv2d_172[0][0]                 
__________________________________________________________________________________________________
conv2d_171 (Conv2D)             (None, 128, 400, 32) 9248        activation_76[0][0]              
__________________________________________________________________________________________________
add_1 (Add)                     (None, 128, 400, 32) 0           batch_normalization_79[0][0]     
                                                                 conv2d_171[0][0]                 
__________________________________________________________________________________________________
batch_normalization_80 (BatchNo (None, 128, 400, 32) 128         add_1[0][0]                      
__________________________________________________________________________________________________
activation_77 (Activation)      (None, 128, 400, 32) 0           batch_normalization_80[0][0]     
__________________________________________________________________________________________________
conv2d_173 (Conv2D)             (None, 64, 200, 64)  18496       activation_77[0][0]              
__________________________________________________________________________________________________
batch_normalization_81 (BatchNo (None, 64, 200, 64)  256         conv2d_173[0][0]                 
__________________________________________________________________________________________________
conv2d_175 (Conv2D)             (None, 64, 200, 64)  18496       add_1[0][0]                      
__________________________________________________________________________________________________
activation_78 (Activation)      (None, 64, 200, 64)  0           batch_normalization_81[0][0]     
__________________________________________________________________________________________________
batch_normalization_82 (BatchNo (None, 64, 200, 64)  256         conv2d_175[0][0]                 
__________________________________________________________________________________________________
conv2d_174 (Conv2D)             (None, 64, 200, 64)  36928       activation_78[0][0]              
__________________________________________________________________________________________________
add_2 (Add)                     (None, 64, 200, 64)  0           batch_normalization_82[0][0]     
                                                                 conv2d_174[0][0]                 
__________________________________________________________________________________________________
batch_normalization_83 (BatchNo (None, 64, 200, 64)  256         add_2[0][0]                      
__________________________________________________________________________________________________
activation_79 (Activation)      (None, 64, 200, 64)  0           batch_normalization_83[0][0]     
__________________________________________________________________________________________________
conv2d_176 (Conv2D)             (None, 32, 100, 128) 73856       activation_79[0][0]              
__________________________________________________________________________________________________
batch_normalization_84 (BatchNo (None, 32, 100, 128) 512         conv2d_176[0][0]                 
__________________________________________________________________________________________________
conv2d_178 (Conv2D)             (None, 32, 100, 128) 73856       add_2[0][0]                      
__________________________________________________________________________________________________
activation_80 (Activation)      (None, 32, 100, 128) 0           batch_normalization_84[0][0]     
__________________________________________________________________________________________________
batch_normalization_85 (BatchNo (None, 32, 100, 128) 512         conv2d_178[0][0]                 
__________________________________________________________________________________________________
conv2d_177 (Conv2D)             (None, 32, 100, 128) 147584      activation_80[0][0]              
__________________________________________________________________________________________________
add_3 (Add)                     (None, 32, 100, 128) 0           batch_normalization_85[0][0]     
                                                                 conv2d_177[0][0]                 
__________________________________________________________________________________________________
batch_normalization_86 (BatchNo (None, 32, 100, 128) 512         add_3[0][0]                      
__________________________________________________________________________________________________
activation_81 (Activation)      (None, 32, 100, 128) 0           batch_normalization_86[0][0]     
__________________________________________________________________________________________________
conv2d_179 (Conv2D)             (None, 16, 50, 256)  295168      activation_81[0][0]              
__________________________________________________________________________________________________
batch_normalization_87 (BatchNo (None, 16, 50, 256)  1024        conv2d_179[0][0]                 
__________________________________________________________________________________________________
conv2d_181 (Conv2D)             (None, 16, 50, 256)  295168      add_3[0][0]                      
__________________________________________________________________________________________________
activation_82 (Activation)      (None, 16, 50, 256)  0           batch_normalization_87[0][0]     
__________________________________________________________________________________________________
batch_normalization_88 (BatchNo (None, 16, 50, 256)  1024        conv2d_181[0][0]                 
__________________________________________________________________________________________________
conv2d_180 (Conv2D)             (None, 16, 50, 256)  590080      activation_82[0][0]              
__________________________________________________________________________________________________
add_4 (Add)                     (None, 16, 50, 256)  0           batch_normalization_88[0][0]     
                                                                 conv2d_180[0][0]                 
__________________________________________________________________________________________________
batch_normalization_89 (BatchNo (None, 16, 50, 256)  1024        add_4[0][0]                      
__________________________________________________________________________________________________
activation_83 (Activation)      (None, 16, 50, 256)  0           batch_normalization_89[0][0]     
__________________________________________________________________________________________________
conv2d_182 (Conv2D)             (None, 16, 50, 256)  590080      activation_83[0][0]              
__________________________________________________________________________________________________
batch_normalization_90 (BatchNo (None, 16, 50, 256)  1024        conv2d_182[0][0]                 
__________________________________________________________________________________________________
activation_84 (Activation)      (None, 16, 50, 256)  0           batch_normalization_90[0][0]     
__________________________________________________________________________________________________
conv2d_183 (Conv2D)             (None, 16, 50, 256)  590080      activation_84[0][0]              
__________________________________________________________________________________________________
up_sampling2d (UpSampling2D)    (None, 32, 100, 256) 0           conv2d_183[0][0]                 
__________________________________________________________________________________________________
concatenate_36 (Concatenate)    (None, 32, 100, 384) 0           up_sampling2d[0][0]              
                                                                 add_3[0][0]                      
__________________________________________________________________________________________________
batch_normalization_91 (BatchNo (None, 32, 100, 384) 1536        concatenate_36[0][0]             
__________________________________________________________________________________________________
activation_85 (Activation)      (None, 32, 100, 384) 0           batch_normalization_91[0][0]     
__________________________________________________________________________________________________
conv2d_184 (Conv2D)             (None, 32, 100, 256) 884992      activation_85[0][0]              
__________________________________________________________________________________________________
batch_normalization_92 (BatchNo (None, 32, 100, 256) 1024        conv2d_184[0][0]                 
__________________________________________________________________________________________________
conv2d_186 (Conv2D)             (None, 32, 100, 256) 884992      concatenate_36[0][0]             
__________________________________________________________________________________________________
activation_86 (Activation)      (None, 32, 100, 256) 0           batch_normalization_92[0][0]     
__________________________________________________________________________________________________
batch_normalization_93 (BatchNo (None, 32, 100, 256) 1024        conv2d_186[0][0]                 
__________________________________________________________________________________________________
conv2d_185 (Conv2D)             (None, 32, 100, 256) 590080      activation_86[0][0]              
__________________________________________________________________________________________________
add_5 (Add)                     (None, 32, 100, 256) 0           batch_normalization_93[0][0]     
                                                                 conv2d_185[0][0]                 
__________________________________________________________________________________________________
up_sampling2d_1 (UpSampling2D)  (None, 64, 200, 256) 0           add_5[0][0]                      
__________________________________________________________________________________________________
concatenate_37 (Concatenate)    (None, 64, 200, 320) 0           up_sampling2d_1[0][0]            
                                                                 add_2[0][0]                      
__________________________________________________________________________________________________
batch_normalization_94 (BatchNo (None, 64, 200, 320) 1280        concatenate_37[0][0]             
__________________________________________________________________________________________________
activation_87 (Activation)      (None, 64, 200, 320) 0           batch_normalization_94[0][0]     
__________________________________________________________________________________________________
conv2d_187 (Conv2D)             (None, 64, 200, 128) 368768      activation_87[0][0]              
__________________________________________________________________________________________________
batch_normalization_95 (BatchNo (None, 64, 200, 128) 512         conv2d_187[0][0]                 
__________________________________________________________________________________________________
conv2d_189 (Conv2D)             (None, 64, 200, 128) 368768      concatenate_37[0][0]             
__________________________________________________________________________________________________
activation_88 (Activation)      (None, 64, 200, 128) 0           batch_normalization_95[0][0]     
__________________________________________________________________________________________________
batch_normalization_96 (BatchNo (None, 64, 200, 128) 512         conv2d_189[0][0]                 
__________________________________________________________________________________________________
conv2d_188 (Conv2D)             (None, 64, 200, 128) 147584      activation_88[0][0]              
__________________________________________________________________________________________________
add_6 (Add)                     (None, 64, 200, 128) 0           batch_normalization_96[0][0]     
                                                                 conv2d_188[0][0]                 
__________________________________________________________________________________________________
up_sampling2d_2 (UpSampling2D)  (None, 128, 400, 128 0           add_6[0][0]                      
__________________________________________________________________________________________________
concatenate_38 (Concatenate)    (None, 128, 400, 160 0           up_sampling2d_2[0][0]            
                                                                 add_1[0][0]                      
__________________________________________________________________________________________________
batch_normalization_97 (BatchNo (None, 128, 400, 160 640         concatenate_38[0][0]             
__________________________________________________________________________________________________
activation_89 (Activation)      (None, 128, 400, 160 0           batch_normalization_97[0][0]     
__________________________________________________________________________________________________
conv2d_190 (Conv2D)             (None, 128, 400, 64) 92224       activation_89[0][0]              
__________________________________________________________________________________________________
batch_normalization_98 (BatchNo (None, 128, 400, 64) 256         conv2d_190[0][0]                 
__________________________________________________________________________________________________
conv2d_192 (Conv2D)             (None, 128, 400, 64) 92224       concatenate_38[0][0]             
__________________________________________________________________________________________________
activation_90 (Activation)      (None, 128, 400, 64) 0           batch_normalization_98[0][0]     
__________________________________________________________________________________________________
batch_normalization_99 (BatchNo (None, 128, 400, 64) 256         conv2d_192[0][0]                 
__________________________________________________________________________________________________
conv2d_191 (Conv2D)             (None, 128, 400, 64) 36928       activation_90[0][0]              
__________________________________________________________________________________________________
add_7 (Add)                     (None, 128, 400, 64) 0           batch_normalization_99[0][0]     
                                                                 conv2d_191[0][0]                 
__________________________________________________________________________________________________
up_sampling2d_3 (UpSampling2D)  (None, 256, 800, 64) 0           add_7[0][0]                      
__________________________________________________________________________________________________
concatenate_39 (Concatenate)    (None, 256, 800, 80) 0           up_sampling2d_3[0][0]            
                                                                 add[0][0]                        
__________________________________________________________________________________________________
batch_normalization_100 (BatchN (None, 256, 800, 80) 320         concatenate_39[0][0]             
__________________________________________________________________________________________________
activation_91 (Activation)      (None, 256, 800, 80) 0           batch_normalization_100[0][0]    
__________________________________________________________________________________________________
conv2d_193 (Conv2D)             (None, 256, 800, 32) 23072       activation_91[0][0]              
__________________________________________________________________________________________________
batch_normalization_101 (BatchN (None, 256, 800, 32) 128         conv2d_193[0][0]                 
__________________________________________________________________________________________________
conv2d_195 (Conv2D)             (None, 256, 800, 32) 23072       concatenate_39[0][0]             
__________________________________________________________________________________________________
activation_92 (Activation)      (None, 256, 800, 32) 0           batch_normalization_101[0][0]    
__________________________________________________________________________________________________
batch_normalization_102 (BatchN (None, 256, 800, 32) 128         conv2d_195[0][0]                 
__________________________________________________________________________________________________
conv2d_194 (Conv2D)             (None, 256, 800, 32) 9248        activation_92[0][0]              
__________________________________________________________________________________________________
add_8 (Add)                     (None, 256, 800, 32) 0           batch_normalization_102[0][0]    
                                                                 conv2d_194[0][0]                 
__________________________________________________________________________________________________
conv2d_196 (Conv2D)             (None, 256, 800, 4)  132         add_8[0][0]                      
==================================================================================================
Total params: 6,287,508
Trainable params: 6,280,212
Non-trainable params: 7,296
__________________________________________________________________________________________________
到达起点
关注
  • 4
    点赞
  • 踩
  • 71
    收藏
    觉得还不错? 一键收藏
  • 16
    评论
专栏目录
Unet、AttentionUnet、R2Unet、R2AUet，可运行，包含数据集
03-24
构建Unet、AttentionUnet、R2Unet、R2AUet，做皮肤病的分割，使用的数据集是ISIC 2017。
ResUNet原理与实现
mefocus的博客
03-13 1万+
U-Net的下采样和上采样的设计与许多现在成熟的网络结构异曲同工。因为ImageNet是一个大规模的图像分类数据集，ImageNet预训练权重可以帮助我们在U-Net的训练中使用更好的初始化权重，加快网络的收敛速度并提高网络的泛化能力。但是，U-Net的下采样和上采样的设计思路和现在许多成熟的网络结构相似，因此，可以看作是先驱性的工作。整个网络结构包括了ResNet的残差连接和U-Net的上下采样和跳跃连接思想，可以更好地平衡特征的丰富性和细节的保留性，在图像分割任务中表现出较好的性能。
深度学习| ResNet和ResUnet
最新发布
qq_40306845的博客
03-18 1318
快速理解ResNet和RestUnet。
【干货教学】unet进阶，如何在unet中加入resnet（残差连接）
BluErroR的博客
11-25 4487
在本教程中，我们将探讨如何在经典的U-Net架构中融入ResNet的残差连接。这种结合了U-Net在图像分割领域的优势和ResNet的残差连接的混合模型，我们称之为ResUnet，旨在通过残差学习改善模型的训练效率和性能。
[深度学习] TensorFlow上实现Unet网络
小黑_BUPT的博客
10-11 2万+
代码取自于 https://github.com/jakeret/tf_unet TensorFlow Unet文档 https://tf-unet.readthedocs.io/en/latest/installation.html TensorFlow Unet安装 确保Tensorflow已安装，如果没有，请参考Tensorflow安装说明 link 克隆github工程 git cl...
ResUnet++
weixin_43921949的博客
02-04 3763
ResUnet++
【医学图像分割网络】之Res U-Net网络PyTorch复现
qq_39071739的博客
06-20 1万+
【医学图像分割网络】之Res U-Net网络PyTorch复现 1.内容 U-Net网络算是医学图像分割领域的开山之作，我接触深度学习到现在大概将近大半年时间，看到了很多基于U-Net网络的变体，后续也会继续和大家一起分享学习。这次分享ResNet+U-Net的一个改进版，ResNet本身就是一个十分优秀的backbone，目前应用仍然十分广泛，我们融合进ResNet后，我们就可以使得U-Net进行迁移学习了。 1.将Resnet作为encoder替换U-Net原始结构 2.U-Net提出时间较早，当时还没
pytorch实现ResNet
热门推荐
winycg的博客
01-30 4万+
参考博客：http://www.cnblogs.com/wuliytTaotao/archive/2018/09/15/9560205.html https://blog.csdn.net/wu_x_j_/article/details/84823002 https://blog.csdn.net/wsp_1138886114/article/details/82080881 残差网络(resid...
SE-ResUNet论文解读
lianghe77的博客
01-14 6528
《Deep Learning Prediction of Incoming Rainfalls：An Operational Service for the City of Beijing China》论文解读 这是一篇通过扩展FCN结构来预测降水的文章。文章提出的服务主要运用于北京区域。它能对未来两小时内的降水进行预测，时间和空间分辨率分别为1km和6分钟。该神经网络模型结合了已知的U-Net，ResNet，Sqeeze-and-Excitation和注意力机制模块。这种方法得到的结果要好于传统方法，且
ResNet 论文总结
Kula的博客
02-24 1250
（一）论文翻译（二）论文笔记（三）代码实现
Hnn-Hnn-MC_3D_ResUnet-64-0.8342-0.8372.hdf5_val_metrics.mat
12-17
Hnn-Hnn-MC_3D_ResUnet-64-0.8342-0.8372.hdf5_val_metrics.mat
Road-Extraction-with-ResUnet
04-13
基于ResUnet模型的Keras道路检测图像分割 该项目旨在借助ResUnet图像分割模型来检测和提取卫星图像中的道路。 ResUnet模型的所有代码都是用Jupyter笔记本编写的，C ++代码用于准备训练过程的数据集，一个python文件用于去除二值化蒙版中的噪声。 图书馆 凯拉斯 皮尔 脾气暴躁的 OpenCV H5py Matplotlib 模型 RESUNET是指深度残留UNET。 这是Zhengxin Zhang等人开发的编码器-解码器体系结构。 用于语义分割。 研究人员采用了它的多种应用，例如息肉分割，脑肿瘤分割，人像分割等等。 RESUNET是一个全卷积神经网络，旨在通过较少的参数获得高性能。 它是对现有UNET体系结构的改进。 RESUNET充分利用了UNET架构和Deep Residual Learning的优势。 数据集 可以通过以下链接下载数据集： :
keras-unet-collection:Tensorflow，Keras U-net，V-net，U-net ++，R2U-net，Attention U-net，ResUnet-a，U ^ 2-Net和UNET 3+的实现，带有受ImageNet训练的主干
05-07
keras-unet-collection 所述tensorflow.keras实施U型网，V-净，U-净++，R2U网，注意力U形网，ResUnet-A，U ^ 2-Net和UNET 3+具有可选ImageNet训练有素骨架。 keras_unet_collection.models包含使用超参数选项配置keras模型的函数。 U-net，U-net ++，Attention U-net和UNET 3+支持预训练的ImageNet主干。 U-net ++，UNET 3+和U ^ 2-Net支持深度监督。 有关其他选项和用例，请参见《 》。 keras_unet_collection.models 名称 参考 unet_2d 网络 vnet_2d V-net（为2-d输入修改） unet_plus_2d U网++ r2_unet_2d R2U网 att
matlab图像分割肿瘤代码-brain_tumor_segmentation:在Matlab中使用ResUNet进行脑肿瘤分割
05-22
matlab图像分割肿瘤代码自述文件 在Matlab中使用ResUNet进行脑肿瘤分割 数据源：脑部MRI分割，将数据粘贴到source文件夹。 运行安装程序以初始化路径。 LGG细分数据集 该数据集包含脑部MR图像以及手动FLAIR异常分割蒙版。 这些图像是从The Cancer Imaging Archive（TCIA）获得的。 他们对应于癌症基因组图谱（TCGA）低级神经胶质瘤收集物中的110例患者，至少具有液衰减倒置恢复（FLAIR）序列和可用的基因组数据。 肿瘤基因组和患者数据在data.csv文件中提供。 所有图像均以.tif格式提供，每个图像有3个通道。 对于101种情况，有3个序列可用，即对比前，FLAIR，对比后（按通道顺序）。 对于9例，缺少造影剂后顺序，对于6例，缺少造影剂前顺序。 丢失的序列将替换为FLAIR序列，以使所有图像变为3通道。 遮罩是二进制的1通道图像。 它们将出现在FLAIR序列中的FLAIR异常分段（适用于所有情况）。 数据集被组织成110个文件夹，每个文件夹都以案例ID命名，其中包含有关源机构的信息。 每个文件夹包含具有以下命名约定的MR图像：
tf.data.Dataset关于batch，repeat，shuffle的讲解
m0_37663944的博客
01-05 5502
本篇博客主要讲解tf.data.Dataset的batch，repeat，shuffle函数，最后着重讲它们之间的顺序使用问题。 tf.data.Dataset官方文档 batch函数讲解 以下先创建一个Dataset类型 import tensorflow as tf t = tf.range(10.)[:, None] t = tf.data.Dataset.from_tensor_slice...
UNet的tensorflow2实现
m0_37663944的博客
01-08 3326
  本篇博客主要使用tensorflow2实现UNet。
DenseNet学习与tensorflow2实现
m0_37663944的博客
02-28 1873
《Densely Connected Convolutional Networks》提出了DenseNet，它用前馈的方式连接每一层与所有其他层，网络的核心结构为如下所示的Dense块，在每一个Dense块中，存在多个Dense层，即下图所示的H1～H4。各Dense层之间彼此均相互连接。 不一定需要在Dense块中对每一个Dense层均直接进行相互连接，来缩小网络的结构；也可能可以在不相邻的D...
resUnet代码实现pytorch
05-24
下面是一个简单的ResUNet实现，使用PyTorch框架： ```python import torch import torch.nn as nn class DoubleConv(nn.Module): def __init__(self, in_ch, out_ch): super(DoubleConv, self).__init__() self.conv = nn.Sequential( nn.Conv2d(in_ch, out_ch, 3, padding=1), nn.BatchNorm2d(out_ch), nn.ReLU(inplace=True), nn.Conv2d(out_ch, out_ch, 3, padding=1), nn.BatchNorm2d(out_ch), nn.ReLU(inplace=True) ) def forward(self, x): x = self.conv(x) return x class Down(nn.Module): def __init__(self, in_ch, out_ch): super(Down, self).__init__() self.mpconv = nn.Sequential( nn.MaxPool2d(2), DoubleConv(in_ch, out_ch) ) def forward(self, x): x = self.mpconv(x) return x class Up(nn.Module): def __init__(self, in_ch, out_ch, bilinear=True): super(Up, self).__init__() if bilinear: self.up = nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True) else: self.up = nn.ConvTranspose2d(in_ch//2, in_ch//2, 2, stride=2) self.conv = DoubleConv(in_ch, out_ch) def forward(self, x1, x2): x1 = self.up(x1) diffY = x2.size()[2] - x1.size()[2] diffX = x2.size()[3] - x1.size()[3] x1 = F.pad(x1, (diffX // 2, diffX - diffX // 2, diffY // 2, diffY - diffY // 2)) x = torch.cat([x2, x1], dim=1) x = self.conv(x) return x class ResUNet(nn.Module): def __init__(self, n_channels, n_classes, bilinear=True): super(ResUNet, self).__init__() self.n_channels = n_channels self.n_classes = n_classes self.bilinear = bilinear self.inc = DoubleConv(n_channels, 64) self.down1 = Down(64, 128) self.down2 = Down(128, 256) self.down3 = Down(256, 512) self.down4 = Down(512, 512) self.up1 = Up(1024, 256, bilinear) self.up2 = Up(512, 128, bilinear) self.up3 = Up(256, 64, bilinear) self.up4 = Up(128, 64, bilinear) self.outc = nn.Conv2d(64, n_classes, 1) def forward(self, x): x1 = self.inc(x) x2 = self.down1(x1) x3 = self.down2(x2) x4 = self.down3(x3) x5 = self.down4(x4) x = self.up1(x5, x4) x = self.up2(x, x3) x = self.up3(x, x2) x = self.up4(x, x1) logits = self.outc(x) return logits ``` 这是一个简单的ResUNet，包含了双卷积，下采样，上采样和输出层。你可以通过改变这些模块的数量和参数来构建一个更深或更浅的网络。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 16
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值