一种基于改进yolov8与知识蒸馏的野生动物目标检测方法

    公开(公告)号:CN118823736A

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

    申请号:CN202410914196.9

    申请日:2024-07-09

    Abstract: 本发明属于计算机视觉目标检测识别与野生动物保护技术领域,具体是一种基于改进yolov8与知识蒸馏技术的野生动物目标检测方法。本发明涉及的技术要点包括:(1)改进原有的yolov8模型,训练一个性能优越的模型作为知识蒸馏的教师模型,再通过改进原始的yolov8模型使其变得轻量化作为知识蒸馏的学生模型。最后,采用logits蒸馏的方法得到一个比原始yolov8模型精度更高,更轻量,性能更优越的模型,并将其部署在国产开发板AXera‑Pi上进行推理验证。(2)教师模型的改进主要包括将原始yolov8网络的backbone替换为Swin Transformer作为主干网络;在yolov8网络模型的neck网络中添加SEAM注意力机制,使模型更有效地处理遮挡场景,提高模型精度。(3)学生模型的改进主要包括将原始的yolov8网络的backbone替换为GhostHGNETV2作为骨干网络;对yolov8的检测头进行重设计,提出Efficient_Detect检测头,使模型网络变得更加轻量化。

Patent Agency Ranking