摘要
本发明公开了一种针对小目标的轻量化检测方法及系统。本发明方法包括以下步骤:获取拍摄小目标历史图像数据,得到图像数据集;对图像数据集进行标注,得到训练数据集;基于YOLOv8n,构建初始轻量化小目标检测模型;S4.使用训练数据集对初始化轻量化小目标检测模型进行训练,得到轻量化小目标检测模型;使用得到的轻量化小目标检测模型,完成实际的针对小目标的轻量化检测。本发明方法基于YOLOv8n构建模型,模型设计了多尺度特征提取模块OMB以及第一空间和通道重建卷积瓶颈特征提取模块(SCBM),降低了模型运行时的计算量和参数量,同时提高了对小目标的检测精度。
技术关键词
网络模块
分支
输出特征
模型训练模块
数据处理模块
上采样
归一化模块
图像
检测头
特征提取模块
数据获取模块
输出模块
积层
特征金字塔网络
卷积特征提取
动态
超参数
系统为您推荐了相关专利信息
企业设备
数据处理方式
数据采集设备
模型更新
变量
盾构掘进参数
等级识别方法
半监督算法
无标签数据
特征提取方法
合成孔径雷达图像
合成孔径雷达SAR图像
训练集
注意力机制
阶段