基于改进型YOLOv5的小目标检测方法及装置
Abstract:
本发明涉及一种基于改进型YOLOv5的小目标检测方法及装置,所述方法包括对YOLOv5网络模型进行改进,得到改进型YOLOv5网络模型;获取小目标图像数据集,并将图像数据集划分为训练样本集和测试样本集;将训练样本集输入至改进型YOLOv5网络模型中进行训练,得到预训练权重,根据预训练权重对改进型YOLOv5网络模型进行调整;将测试样本集输入调整后的改进型YOLOv5网络模型中,得到检测结果。本发明通过对现有的YOLOv5网络模型进行改进,提高了检测工作的准确性、即时性,减少了人力资源损耗,有效提高监测效率。
Patent Agency Ranking
0/0