一种改进的YOLOv4-Tiny实现工程机械及物料的识别方法
摘要:
本发明涉及人工智能的领域,公开了一种改进的YOLOv4‑Tiny实现工程机械及物料的识别方法,包括:采集工程机械及物料图片,建立数据集样本,数据集样本用于训练改进的YOLOv4‑Tiny模型;建立改进的YOLOv4‑Tiny模型,在原YOLOv4‑Tiny网络的主干特征提取网络最后加入SPP多尺度特征融合模块,以实现感受野的扩大,在SPP多尺度特征融合模块前插入SE通道注意力模块,以实现无关信息干扰的减少,调整FPN特征金字塔融合模块,加强对浅层信息的利用,以实现网络表征能力的增强;基于改进的YOLOv4‑Tiny模型,使用数据集样本进行模型训练,以实现对工程机械及物料目标的检测与分类。实验结果表明,本发明的算法能使工程机械以及物料的检测与分类更准确。
0/0