科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于YOLO-Terse网络的目标检测方法,包括：获取包含待检测目标的待检测图像；将待检测图像输入预先训练的YOLO-Terse网络,根据待检测图像的特征确定待检测目标所属的类别,以及待检测目标在待检测图像中位置；其中,YOLO-Terse网络是在YOLOv3网络基础上,采用层级和通道级剪枝,再结合知识蒸馏引导网络恢复后形成的。本发明通过对YOLOv3进行层剪枝、稀疏化训练、通道剪枝,以及知识蒸馏处理,并选取优化的处理参数,得到精简的YOLO-Terse网络,该网络体积大幅缩小,摒除了大部分的冗余计算,基于该网络的目标检测速度得到大幅度的提升,而且能够维持检测的精度。
- 单位
