Research of Small Object Detection Problem Based on MDS-YOLO Model
To solve the problem of large computation and low accuracy of the current mainstream algorithms for small object detection,this paper replaces the backbone network in YOLOv4 with the lightweight network MobileNetV3,and replaces some ordinary convolutions in the neck network with depthwise separable convolutions.At the same time,a new loss function IF-EIoU Loss is defined for small object detection.Therefore,MDS-YOLO object detection model is constructed.This model has a high detection speed and good detection performance for small object.To verify the effectiveness of the model,experiments are carried out on MS COCO dataset and Visdrone2019 dataset,respectively.Compared with the YOLOv4 algorithm,on MS COCO dataset,the average detection accuracy of the MDS-YOLO algorithm is improved by 1.5 percentage points,the detection accuracy of small object is increased by 3.3 percentage points,and the detection speed is also increased from 31 frames per second to 36 frames per second.On the Visdrone2019 dataset,the MDS-YOLO algorithm increases the average detection accuracy from 14.9%of YOLOv4 to 16.3%.The experimental results show that the MDS-YOLO algorithm proposed can effectively improve the detection accuracy of small object.
small object detectionYOLOv4 algorithmlightweight network MobileNetV3IF-EIoU LossMS COCO dataset
朱恩文、梁曌、肖进文、梁小林
展开 >
长沙理工大学 数学与统计学院,湖南 长沙,410114
湖南工程学院 计算科学与电子学院,湖南 湘潭,411104
小目标检测 YOLOv4算法 轻量级网络MobileNetV3 IF-EIoU Loss MS COCO数据集
国家科技期刊平台
NSTL
万方数据
