一种目标检测网络的压缩方法、装置、设备及介质

    公开(公告)号:CN117744715A

    公开(公告)日:2024-03-22

    申请号:CN202410009884.0

    申请日:2024-01-03

    Abstract: 本发明公开了一种目标检测网络的压缩方法、装置、设备及介质。目标检测网络的压缩方法,包括:基于跨阶段局部网络,创建第一目标检测网络,并对第一目标检测网络进行残差层扩展,得到第二目标检测网络;获取目标算子的结构参数、网络预训练参数以及标注样本集;根据标注样本集、第二目标检测网络以及目标损失函数,优化目标算子的结构参数以及网络预训练参数,得到目标结构参数以及目标网络参数;根据目标结构参数压缩第二目标检测网络,并进行通道剪枝处理,得到目标待训练网络;基于标注样本集以及目标待训练网络,确定目标轻量网络。本发明实施例的技术方案能够提升目标检测网络的网络压缩效果,从而提高网络的计算速度。

Patent Agency Ranking