骨干网络微调剪枝方法、设备、存储介质及产品
摘要:
本发明公开了一种骨干网络微调剪枝方法、设备、存储介质及产品,所述方法包括获取未剪枝骨干网络,并对未剪枝骨干网络进行剪枝处理,得到剪枝骨干网络;生成合成图像;将合成图像输入至未剪枝骨干网络,以未剪枝骨干网络的输出特征图作为合成图像的伪标签;根据合成图像及其伪标签构建合成数据集;利用合成数据集对所述剪枝骨干网络进行训练,得到目标骨干网络;获取头部网络,构建与特定任务对应的样本数据集;利用样本数据集对头部网络进行训练,得到目标头部网络;对目标骨干网络和目标头部网络进行组合,得到特定任务模型。本发明无需采用额外生成器网络来生成合成图像,且通过损失值的反向传播保证了合成图像的生成质量。
0/0