一种基于改进YOLOV9模型的电力设备目标检测方法和装置

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于改进YOLOV9模型的电力设备目标检测方法和装置
申请号:CN202411039403
申请日期:2024-07-31
公开号:CN118710926A
公开日期:2024-09-27
类型:发明专利
摘要
本发明提供了一种基于改进YOLOV9模型的电力设备目标检测方法和装置,属于目标检测技术领域。方法包括:获取电力设备的红外图像和可见光图像保存为数据集;使用数据增强技术对所述数据集进行扩充;采用Labelimg工具对扩充后的数据集进行标注划分为训练集、验证集和测试集;通过在原始YOLOV9神经网络模型中引入SPDConv模块、PPMDC模块和DASI模块构建改进的YOLOV9神经网络模型为YOLOV9s‑Improve神经网络模型;通过训练集对所述YOLOV9s‑Improve神经网络模型进行训练,迭代更新参数生成训练后的YOLOV9s‑Improve神经网络模型;将实时获取的电力设备的红外图像和可见光图像输入至训练后的YOLOV9s‑Improve神经网络模型中生成电力设备的检测结果。通过对原始YOLOV9神经网络模型的整体结构进行改进,优化了模型对小目标和低分辨率图像的处理能力。
技术关键词
神经网络模型 电力设备 可见光图像 训练集 数据 图片 辅助检测头 样本 信息模块 图像特征信息 多尺度特征 分支 尺寸 坐标 注意力 生成特征 参数