← 返回列表
一种基于轻量化网络模型的图像分类方法及搭载装置
申请人信息
- 申请人:广东石油化工学院
- 申请人地址:525099 广东省茂名市官渡二路139号大院
- 发明人: 广东石油化工学院
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 一种基于轻量化网络模型的图像分类方法及搭载装置 |
| 专利类型 | 发明申请 |
| 申请号 | CN202311452715.6 |
| 申请日 | 2023/11/3 |
| 公告号 | CN117475219A |
| 公开日 | 2024/1/30 |
| IPC主分类号 | G06V10/764 |
| 权利人 | 广东石油化工学院 |
| 发明人 | 刘世杰; 刘美; 杨涛 |
| 地址 | 广东省茂名市茂南区官渡二路139号 |
摘要文本
广东石油化工学院取得“一种透气窗帘布”专利技术,本发明公开了一种基于轻量化网络模型的图像分类方法及搭载装置,包括以下步骤:构建轻量化网络模型;基于秃鹰搜索算法优化所述轻量化网络模型的学习率,获得优化后的轻量化网络模型;在知识蒸馏模型中引入所述优化后的轻量化网络模型作为助手网络,然后基于秃鹰搜索算法对所述知识蒸馏模型的温度进行优化,获得优化后的知识蒸馏模型;基于所述优化后的知识蒸馏模型对所述轻量化网络模型进行蒸馏,获得图像分类模型;对所述图像分类模型进行训练,基于训练后的图像分类模型,对肺部图像进行识别分类。本发明将训练后的图像分类模型搭载在实时处理装置上,通过该装置改进的模型加速电路使其推理速度和处理速度都有明显提升,且功耗降低。
专利主权项内容
1.一种基于轻量化网络模型的图像分类方法,其特征在于,包括以下步骤:构建轻量化网络模型;基于秃鹰搜索算法优化所述轻量化网络模型的学习率,获得优化后的轻量化网络模型;在知识蒸馏模型中引入所述优化后的轻量化网络模型作为助手网络,然后基于秃鹰搜索算法对所述知识蒸馏模型的温度进行优化,获得优化后的知识蒸馏模型;基于所述优化后的知识蒸馏模型对所述轻量化网络模型进行蒸馏,获得图像分类模型;对所述图像分类模型进行训练,基于训练后的图像分类模型,对肺部图像进行识别分类。