← 返回列表
基于Transformer蒸馏的仪表检测方法、系统、设备及介质
申请人信息
- 申请人:佳源科技股份有限公司
- 申请人地址:210000 江苏省南京市雨花台区宁双路19号云密城14-17层
- 发明人: 佳源科技股份有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 基于Transformer蒸馏的仪表检测方法、系统、设备及介质 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311284014.6 |
| 申请日 | 2023/10/7 |
| 公告号 | CN117557824A |
| 公开日 | 2024/2/13 |
| IPC主分类号 | G06V10/764 |
| 权利人 | 佳源科技股份有限公司 |
| 发明人 | 张言; 戴慧丹; 沈阳 |
| 地址 | 江苏省南京市雨花台区宁双路19号云密城14-17层 |
摘要文本
本发明涉及一种基于Transformer蒸馏的仪表检测方法、系统、计算机设备及存储介质。所述方法包括:基于DCGAN模型对仪表图像进行图像预处理,输出得到目标仪表图像;通过Diffusion Model网络模型对目标仪表图像进行数据增强处理,并从目标仪表图像中生成具有分类特征的训练样本;利用Transformer进行蒸馏,以CenterNet网络作为教师网络,以CNN网络作为学生网络,训练得到目标检测网络;通过目标检测网络进行仪表检测,得到检测结果。基于DCGAN模型、Diffusion Model网络模型进行图像去噪处理和数据增强,可以减少配电站外界环境与随机干扰对图像产生的影响;基于Transformer蒸馏,可以实现模型的轻量化,保证检测准确性,提高了目标的检测精度。。专利查询网
专利主权项内容
(来源 马克数据网) 1.一种基于Transformer蒸馏的仪表检测方法,其特征在于,所述方法包括:采集仪表图像,并基于DCGAN模型对所述仪表图像进行图像预处理,输出得到目标仪表图像;通过Diffusion Model网络模型对所述目标仪表图像进行数据增强处理,并从所述目标仪表图像中生成具有分类特征的训练样本;利用Transformer进行蒸馏,以CenterNet网络作为教师网络,以CNN网络作为学生网络,将所述训练样本作为输入,训练得到目标检测网络;通过所述目标检测网络进行仪表检测,得到检测结果。