发明公开
- 专利标题: 骨干网络微调剪枝方法、设备、存储介质及产品
-
申请号: CN202410846273.1申请日: 2024-06-27
-
公开(公告)号: CN118865051A公开(公告)日: 2024-10-29
- 发明人: 杨启彬 , 胡玉鹏 , 刘邦洪 , 杨德志 , 旷文鑫
- 申请人: 湖南开鸿智谷数字产业发展有限公司
- 申请人地址: 湖南省长沙市高新开发区桐梓坡西路298号湖南拓维信息系统股份有限公司研发大楼三楼南侧
- 专利权人: 湖南开鸿智谷数字产业发展有限公司
- 当前专利权人: 湖南开鸿智谷数字产业发展有限公司
- 当前专利权人地址: 湖南省长沙市高新开发区桐梓坡西路298号湖南拓维信息系统股份有限公司研发大楼三楼南侧
- 代理机构: 长沙正奇专利事务所有限责任公司
- 代理商 李发军; 曾利平
- 主分类号: G06V10/82
- IPC分类号: G06V10/82 ; G06N3/0464 ; G06N3/082 ; G06N3/084
摘要:
本发明公开了一种骨干网络微调剪枝方法、设备、存储介质及产品,所述方法包括获取未剪枝骨干网络,并对未剪枝骨干网络进行剪枝处理,得到剪枝骨干网络;生成合成图像;将合成图像输入至未剪枝骨干网络,以未剪枝骨干网络的输出特征图作为合成图像的伪标签;根据合成图像及其伪标签构建合成数据集;利用合成数据集对所述剪枝骨干网络进行训练,得到目标骨干网络;获取头部网络,构建与特定任务对应的样本数据集;利用样本数据集对头部网络进行训练,得到目标头部网络;对目标骨干网络和目标头部网络进行组合,得到特定任务模型。本发明无需采用额外生成器网络来生成合成图像,且通过损失值的反向传播保证了合成图像的生成质量。