您的位置:
首页
>
农业专利
>
详情页
一种基于HA-UNet网络的眼底图像视盘检测方法
- 专利权人:
- 杭州电子科技大学;中电数据服务有限公司
- 发明人:
- 胡文丽,周晓飞,张继勇,李世锋,周振,何帆
- 申请号:
- CN202211093428.6
- 公开号:
- CN115587967A
- 申请日:
- 2022.09.06
- 申请国别(地区):
- CN
- 年份:
- 2023
- 代理人:
- 摘要:
- 本发明涉及一种基于HA‑UNet网络的眼底图像视盘检测方法,包括如下步骤:数据预处理、模型构建、模型训练和模型评估。所述数据预处理包括对图像的缩放与剪切;所述构建的HA‑UNet网络在原始UNet网络的基础上,采用残差模块替代原UNet中的卷积层,并提出混合注意力模块,即HA模块,建立多注意力机制与特征之间的关系,对前景信息与背景信息进行挖掘与融合,同时,该网络采用混合损失函数,即BCE损失函数、SSIM损失函数和IoU损失函数的结合作为模型最终的损失函数;所述模型训练即当模型的损失函数不再降低之后,保存模型;所述模型评估即将训练好的模型放到测试集上进行评估。
- 来源网站:
- 中国工程科技知识中心
- 来源网址:
- http://www.ckcest.cn/home/