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