《基于改进U-Net网络的气胸分割方法》论文笔记

一.模型设计
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参数作为编码模块的初始参数

猜你喜欢

转载自blog.csdn.net/qq_45234219/article/details/111927609