ABSTRACT
In many applications, existing image detection algorithms detect small targets in complex scenes with low recognition rate and biased classification. To address this issue, we develop a multi-target detection algorithm based on improved Retinanet. It consists of the following four parts: 1) extract rich texture features using the information interaction module; and 2) extract high-level abstract features through the improved FPN+ module; and 3) make full use of contextual information for detection leveraging SSH detection head; and 4) adopt the weighted loss for regression. For the accuracy of large and small categories and avoid bias, we employ the ensemble of loss function to optimize parameters. The experimental results confirm the feasibility of our detection algorithm, with the improved performance on four metrics. The mAP50 value and mmAP value increase to 96.75% and 75.08% respectively, and the ACD value decreases by 0.907.
- AGRAWAL P, GIRSHICK R, MALIK J. Analyzing the performance of Multilayer Neural Networks for Object Recognition[J]. Lecture Notes in Computer Science,2014.Google Scholar
- He K, Zhang X, Ren S, et al.Spatial pyramid pooling in deep convolutional networks for visual recognition[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015, 37)9(:1904-1916.Google ScholarDigital Library
- Ren S, He K, Girshick R, et al.Faster R-CNN:towards real-time object detection with region proposal networks[C]// Conference on Neural Information Processing Systems, 2015:91-99.Google Scholar
- Girshick R.Fast R-CNN[C]//International Conference on Computer Vision,2015:1440-1448.Google Scholar
- Dai J,Li Y,He K,et al.R-FCN:object detection via region-based fully convolutional networks[C]/ /Conference on Neural Information Processing Systems,2016:379-387.Google Scholar
- Lin T Y,Goyal P,Girshick R,et al.Focal loss for dense object detection[C]//International Conference on Computer Vision,2017:2999-3007.Google Scholar
- Duan K,Bai S,Xie L,et al.Center Net:keypoint triplets for object detection[J].ar Xiv:1904.08189,2019.Google Scholar
- Mark Everingham, John Winn. The PASCAL Visual Object Classes Challenge 2012 (VOC2012) Development Kit:,2012.http://host.robots.ox.ac.uk/pascal/VOC/.Google Scholar
- LIN Tsungyi, DOLLÁR P, GIRSHICK R, Feature pyramid networks for object detection[C]// Proceedings of the IEEE Conference on Computer Vision And Pattern Recognition. Honolulu, HI, USA: IEEE, 2017:2117-2125.Google Scholar
- K. He, X. Zhang, S. Ren and J. Sun, "Deep Residual Learning for Image Recognition," 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, 2016, pp. 770-778, doi: 10.1109/CVPR.2016.90.Google Scholar
- T. Lin, P. Dollár, R. Girshick, K. He, B. Hariharan and S. Belongie, "Feature Pyramid Networks for Object Detection," 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, 2017, pp. 936-944, doi: 10.1109/CVPR.2017.106.Google Scholar
- Xiu-Shen Wei, Quan Cui, Lei Yang ,et al. RPC: A Large-Scale Retail Product Checkout Dataset[J]. arXiv:1901.07249v1 [cs.CV] 22 Jan 2019.Google Scholar
Recommendations
Recaptured images detection algorithm based on frequency domain transformation1
With the raising popularity of digital devices in the current society, the present image detection system is becoming a great threaten. Especially the appearance of the recaptured images. It can be used in traditional invalid digital image ...
RETRACTED ARTICLE: Analysis of sports image detection technology based on machine learning
AbstractCurrent sports competitions are mostly broadcast in the form of live video or video files, and information detection for athletes and sports economic processes can also be carried out through image detection technology. However, from the current ...
PCB Defect Detection Method Based on Improved RetinaNet
Machine Learning for Cyber SecurityAbstractIn the electronic industry product quality control, PCB defect detection is a crucial part, which has the characteristics of small defect size and high similarity. The existing defect detection methods are still not good enough for detecting small ...
Comments