当前位置:   article > 正文

《基于改进U-Net网络的气胸分割方法》论文笔记_efficientnet作为unet的编码器

efficientnet作为unet的编码器

一.模型设计
1.用的unet的整体结构
在这里插入图片描述
2.收缩路径中的编码模块使用的时EfficientNet中的MBCconvBlock模块在这里插入图片描述
3.扩展路径中的解码模块是自己设计的
在这里插入图片描述
其中的scSE是将空间注意力机制(sSE)和通道注意力机制(cSE)并联相加,如下图:在这里插入图片描述
在扩展路径中,反卷积的输出层先合并左边编码模块的跳层连接模块,然后经过解码模块,再使用上采样和反卷积到原图大小
二.其他参数
1.损失函数:结合了Dice相似系数损失函数和Focal损失函数其中a是类别权重,用来平衡正负样本的比例不均,经实验设置为10
2.还设计了一个三重阈值策略,用来更精准判断气胸像素
3.评价指标:Dice,Recall,Precision
4.训练集:12089副 ;验证集:3205副;epoch=60;batch_size=2;初始学习率=1e-5;学习率衰减方式:ReduceLROnPlateau
5.使用了数据增强来增强模型的抗过拟合和抗干扰能力
使用了迁移学习,将训练好的EfficientNet参数作为编码模块的初始参数

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/626175
推荐阅读
相关标签
  

闽ICP备14008679号