发明公开
- 专利标题: 一种改进的YOLOv4-Tiny实现工程机械及物料的识别方法
-
申请号: CN202310269243.4申请日: 2023-03-20
-
公开(公告)号: CN116524232A公开(公告)日: 2023-08-01
- 发明人: 艾春 , 王旭东 , 刘淑英 , 高根
- 申请人: 缤谷电力科技(上海)有限公司
- 申请人地址: 上海市嘉定区嘉定工业区叶城路1288号6幢J2236室
- 专利权人: 缤谷电力科技(上海)有限公司
- 当前专利权人: 缤谷电力科技(上海)有限公司
- 当前专利权人地址: 上海市嘉定区嘉定工业区叶城路1288号6幢J2236室
- 代理机构: 上海汉声知识产权代理有限公司
- 代理商 胡晶
- 主分类号: G06V10/764
- IPC分类号: G06V10/764 ; G06V10/774 ; G06V10/82 ; G06V10/80 ; G06N3/0464 ; G06N3/048 ; G06N3/08
摘要:
本发明涉及人工智能的领域,公开了一种改进的YOLOv4‑Tiny实现工程机械及物料的识别方法,包括:采集工程机械及物料图片,建立数据集样本,数据集样本用于训练改进的YOLOv4‑Tiny模型;建立改进的YOLOv4‑Tiny模型,在原YOLOv4‑Tiny网络的主干特征提取网络最后加入SPP多尺度特征融合模块,以实现感受野的扩大,在SPP多尺度特征融合模块前插入SE通道注意力模块,以实现无关信息干扰的减少,调整FPN特征金字塔融合模块,加强对浅层信息的利用,以实现网络表征能力的增强;基于改进的YOLOv4‑Tiny模型,使用数据集样本进行模型训练,以实现对工程机械及物料目标的检测与分类。实验结果表明,本发明的算法能使工程机械以及物料的检测与分类更准确。