基于旋转目标检测的包裹位置实时检测方法、装置、设备和介质

摘要

本发明公开了一种基于旋转目标检测的包裹位置实时检测方法、装置、计算机设备和存储介质，包括：根据获取的实时物流包裹图片得到数据集；根据数据集的特点，设计的旋转目标检测网络模型采用目标检测网络Yolov5作为基础架构，在特征融合Neck中，修改PANet结构为双向特征金字塔的特征融合网络Bi‑FPN，在Bi‑FPN中插入注意力机制，在检测层Head中调整检测头分支数以及增加角度分类输出和角度损失函数，根据增加的角度分类输出预测物流包裹的旋转矩形框；利用数据集对模型进行训练；将获取的视频流输入训练好的模型，根据输出包裹的实时状态信息，实现包裹单件分离。本发明提供的方法，不受包裹尺寸限制，检测精度高。

关键词