
基于Unet的胰腺癌数据分割研究【含数据集、完整代码及训练结果】
5星
- 浏览量: 0
- 大小:None
- 文件类型:7Z
简介:
本研究运用Unet模型进行胰腺癌数据分割,并提供详尽的数据集、完整代码和训练成果展示。适合医学影像分析与深度学习技术结合的研究者参考使用。
本项目基于Unet网络对人胰腺癌症进行分割处理。数据集包括胰腺癌症的图像及其对应的掩码(mask),其中1代表胰腺、2代表癌症区域,0则表示背景。
在训练过程中,模型经过50个epoch后,在全局像素点准确度上达到了99%,miou指标为0.90。进一步延长训练周期可望提升性能表现。
项目代码包含以下部分:
- **训练**:通过train脚本自动进行模型训练,并且数据会被随机缩放至设定尺寸的50%-150%之间,实现多尺度训练效果。在utils模块中的compute_gray函数负责将掩码灰度值保存为文本段落件并定义Unet网络输出通道。
- **介绍**:采用余弦衰减策略调整学习率,并且通过matplotlib库绘制了训练集和测试集中损失及iou的曲线图,这些信息存储于run_results目录内。此外还记录有详细的训练日志、最佳权重等数据,在日志中可以查看各类别的iou值、召回率(recall)、精确度(precision)以及全局像素点准确度。
- **推理**:将待处理图像放置在inference文件夹下,直接运行predict脚本即可进行预测操作。
具体使用方法请参阅README文档。对于初学者而言也非常友好;如果需要训练自己的数据集,请按照指定格式准备相应数据后替换现有内容便可开始实验。
全部评论 (0)
还没有任何评论哟~


