SR-YOLO: Small Objects Detection Based on Super Resolution
Abstract
Since the introduction of convolutional neural networks, object detection based on deep learning has made great progress and has been widely used in the industry. However, because the weak and small object contains too little information, the samples are rich in diversity, and there are different degrees of occlusion, the detection difficulty is too great, and the object detection has entered a bottleneck period of development. We firstly introduce a super-resolution network to solve the problem of small object pixel area being too small, and fuse the super-resolution generator with the object detection baseline model for collaborative training. In addition, in order to reinforce the weak feature of small objects, we design a convolution block based on the edge detection operator Sobel. Experiments show that proposed method achieves mAP50 improvement of 2.4% for all classes and 4.4% for the relatively weak pedestrian class on our dataset relative to the Yolov5s baseline model.