基本信息:
- 专利标题： 一种紫外图像语义分割方法及装置
- 申请号：CN202311421746.5 申请日：2023-10-27
- 公开(公告)号：CN117671253A 公开(公告)日：2024-03-08
- 发明人： 吴军 , 向华桥 , 邓鹤鸣 , 王骞 , 陈雪松 , 陆潇 , 杨世强 , 闻铖 , 陈京晶 , 李季 , 王媛 , 鲁敏
- 申请人： 国网湖北省电力有限公司超高压公司 , 湖北省超能电力有限责任公司 , 湖北大学
- 申请人地址： 湖北省武汉市汉阳区五里墩二合村60-65号
- 专利权人： 国网湖北省电力有限公司超高压公司,湖北省超能电力有限责任公司,湖北大学
- 当前专利权人： 国网湖北省电力有限公司超高压公司,湖北省超能电力有限责任公司,湖北大学
- 当前专利权人地址： 湖北省武汉市汉阳区五里墩二合村60-65号
- 代理机构： 武汉知产时代知识产权代理有限公司
- 代理人： 徐欢
- 主分类号： G06V10/26
- IPC分类号： G06V10/26 ; G06V20/70 ; G06V10/82 ; G06N3/0464 ; G06V10/774 ; G06V10/776 ; G06N3/082 ; G06N3/045 ; G06N3/096
摘要：
本发明公开了一种紫外图像语义分割方法及装置，方法包括步骤：采集紫外图像样本数据；对图像预处理；搭建基于UNet改进的语义分割模型；训练改进的语义分割模型，得到性能最优模型；对性能最优模型进行稀疏训练；采用通道剪枝方法对稀疏训练后的语义分割模型进行压缩剔除网络中大量冗余参数，得到轻量化模型；同时对轻量化模型使用训练集和验证集进行知识蒸馏进行微调训练，并对微调训练后模型进行性能评估测试，得到微调模型；将微调模型的权重文件加载到构建改进的语义分割模型中，并利用该模型对紫外图像进行分割检测。装置对应方法。本发明在减少网络模型计算量及参数量的同时并未导致分割效果下降，对移动终端设备更友好。