目标检测模型的剪枝方法及其装置、电子设备

本发明公开了一种目标检测模型的剪枝方法及其装置、电子设备，涉及模型剪枝技术领域，包括：对各模块结构的掩码进行初始化，得到初始化掩码，并确定各模块结构的重要性得分，对最小重要性得分的模块结构进行剪枝，以根据剩余注意力头集合中各注意力头的注意力头运算量和剩余过滤器集合中各过滤器的过滤器运算量，在运算量条件约束下确定目标剪枝组合，对目标剪枝组合进行剪枝，得到初始剪枝后的目标检测模型，在延迟条件约束下确定初始剪枝后的目标检测模型中每层网络中待剪枝的注意力头和过滤器，并进行剪枝，得到剪枝完成的目标检测模型。本发明解决了相关技术中确定模型中需要剪枝的结构的准确性较低，导致剪枝后的模型的精度较差的技术问题。