一种面向火焰检测的后训练剪枝方法

本发明一种基于梯度加权类激活映射的后训练剪枝模型的火焰识别方法一种面向火焰检测的后训练剪枝方法，包括以下步骤：获取待识别火焰图像；对基于梯度加权类激活映射的后训练剪枝模型进行训练，得到训练好的梯度加权类激活映射的后训练剪枝模型；将待识别火焰图像输入到训练好的梯度加权类激活映射的后训练剪枝模型中，得到火焰的单通道热力图；对火焰的单通道热力图筛选，将筛选后的单通道热力图筛选合成得到最终火焰的热力图；基于梯度加权类激活映射的后训练剪枝方法可以灵活地定义和调整通道重要性评估标准，适应不同的应用场景和需求。