- 专利标题: 一种基于ResCo-UNet的荞麦籽粒图像分割方法
-
申请号: CN202410088595.4申请日: 2024-01-22
-
公开(公告)号: CN117893763B公开(公告)日: 2024-09-10
- 发明人: 吕少中 , 熊政炳
- 申请人: 内蒙古工业大学
- 申请人地址: 内蒙古自治区呼和浩特市新城区爱民街49号
- 专利权人: 内蒙古工业大学
- 当前专利权人: 内蒙古工业大学
- 当前专利权人地址: 内蒙古自治区呼和浩特市新城区爱民街49号
- 代理机构: 北京虑安知识产权代理有限公司
- 代理商 高岩
- 主分类号: G06V10/26
- IPC分类号: G06V10/26 ; G06V10/44 ; G06V10/77 ; G06V10/94 ; G06V10/82 ; G06N3/0455 ; G06N3/045
摘要:
本发明涉及图像分割技术领域,具体涉及一种基于ResCo‑UNet的荞麦籽粒图像分割方法,包括以下步骤:基于传统U‑Net进行改进,构建ResCo‑UNet模型;使用工业相机拍摄采集荞麦籽粒图像,使用改进的卷积生成对抗网络进行数据增广;使用数据标注工具进行图像标注,形成数据集,所述数据集包括训练集、测试集以及验证集;使用训练集训练ResCo‑UNet模型,保留最优模型权重文件;将测试集输入ResCo‑UNet模型,载入最优模型权重文件;使用验证集对ResCo‑UNet模型进行验证有效性。本发明,在传统的U‑Net基础上对编码器和解码器进行了改进,提出在编码器中加入并行的CA2注意力机制,有效地对荞麦籽粒的特征进行提取,在解码器中使用ConvNeXt Block进一步提炼提取的语义特征,使用TensorRT引擎对推流流程进行加速。
公开/授权文献
- CN117893763A 一种基于ResCo-UNet的荞麦籽粒图像分割方法 公开/授权日:2024-04-16