← 返回列表
轻量化多尺度UNet网络的图像分割方法
摘要文本
本发明公开了一种轻量化多尺度UNet网络的图像分割方法，涉及图像处理方法技术领域。所述方法包括如下步骤：数据预处理，对数据集进行数据增强和划分；构建多尺度的轻量化UNet图像分割网络，所述轻量化UNet图像分割网络包括采用变尺度结构的编码器和解码器两部分；将解码器网络输出的网络特征进行处理，输出每个像素对应的概率，并结合均方误差损失函数MSE，对图像分割网络进行训练；对所述图像分割网络输出的单像素概率进行后处理，生成分割图像。所述方法具有更少的参数，同时具有更高的检测精度，能够精准的完成图像的分割。
申请人信息
- 申请人:河北翔拓航空科技有限公司; 石家庄铁道大学
- 申请人地址:050299 河北省石家庄市鹿泉区御园路99号光谷科技园邱氏电子科技园2栋2楼东
- 发明人: 河北翔拓航空科技有限公司; 石家庄铁道大学
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 轻量化多尺度UNet网络的图像分割方法 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311270975.1 |
| 申请日 | 2023/9/28 |
| 公告号 | CN117351202A |
| 公开日 | 2024/1/5 |
| IPC主分类号 | G06V10/26 |
| 权利人 | 河北翔拓航空科技有限公司; 石家庄铁道大学 |
| 发明人 | 赵志宏; 何朋; 郝子晔; 俞阿丹; 李万根 |
| 地址 | 河北省石家庄市鹿泉区御园路99号光谷科技园邱氏电子科技园2栋2楼东; 河北省石家庄市北二环东路17号 |
专利主权项内容
1.一种轻量化多尺度UNet网络的图像分割方法，其特征在于包括如下步骤：S1 : 数据预处理，对数据集进行数据增强和划分；S2：构建多尺度的轻量化UNet图像分割网络，所述轻量化UNet图像分割网络包括采用变尺度结构的编码器和解码器两部分，两部分之间通过两个3×3的卷积连接，使用基于Ghost空洞卷积的GLBM模块作为网络的基本特征提取模块，并在网络中添加EMA注意力机制进一步提升分割目标的关注度；S3：将解码器网络输出的网络特征进行处理，输出每个像素对应的概率，并结合均方误差损失函数(MSE)，对图像分割网络进行训练；S4：对所述图像分割网络输出的单像素概率进行后处理，生成分割图像。。 （更多数据，详见马克数据网）