Electro-Optic Technology Application, Volume. 38, Issue 6, 60(2023)
Improve YOLOv5 Steel Surface Defect Detection Technology
[1] [1] KRIZHEVSKY A, SUTSKEVER I, HINTONGE. Imagenet classification with deep con-volutional neural networks[J]. Advances in Neural Information Processing Systems, 2012, 25(2): 84-90.
[3] [3] REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition. New York: IEEE, 2016: 779-788.
[4] [4] REDMON J, FARHADI A. YOLO9000: better, faster, stronger[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition. New York: IEEE Press, 2017: 6517-6525.
[5] [5] REDMON J, FARHADI A. YOLOv3: an incremental improvement[C]//2018 IEEE Conference on Computer Vision and Pattern Recognition. New York: IEEE, 2018, arXiv: 1804.0276.
[6] [6] BOCHKOVSKIY A, WANG C Y, LIAO H. YOLOv4: optimal speed and accuracy of object detection[J]. Computer Vision and Pattern Recognition, 2020, arXiv preprint arXiv: 2004.10934.
[7] [7] HE K M, ZHANG X, REN S Q, 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.
[10] [10] LIN T Y, DOLLáR P, GIRSHICK R, et al. Feature pyramid networks for object detection[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition(CVPR). Hawaii, USA: IEEE, 2017: 936-944.
[11] [11] LIU S, QI L, QIN H F, et al. Path aggregation network for instance segmentation[C]//2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). New York: IEEE, 2018: 8759-8768.
[12] [12] YU J, JIANG Y, WANG Z, et al. UnitBox: an advanced object detection network[C]//24th ACM International Conference on Multimedia. Amsterdam, The Netherlands, New York: ACM Press, 2016: 516-520.
[13] [13] REZATOFIGHI H, TSOI N, JY G, et al. Generalized intersection over union: a metric and a loss for bounding box regression[C]//2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). New York: IEEE, 2019.
[14] [14] ZHENG Z, WANG P, LIU W, et al. Distance IoU loss: faster and better learning for bounding box regression[C]//Proceeding of the AAAI Conference on Artificial Intelligence. Menlo Park, CA: AAAIPress, 2020: 12993-13000.
[16] [16] WANG Y N, REN J S. Taxi passenger hot spot mining based on a refined K-means++ algorithm[J]. IEEE Access, 2021, 9: 66587-66598.
[17] [17] YUN S, HAN D, OH S J, et al. Cutmix: regularization strategy to train strong classifiers with localizable features[C]//Proceedings of The IEEE/CVF International Conference on Computer Vision. Seoul, Korea: IEEE, 2019: 6023-6032.
Get Citation
Copy Citation Text
LI Dengyue, JIANG Yueqiu. Improve YOLOv5 Steel Surface Defect Detection Technology[J]. Electro-Optic Technology Application, 2023, 38(6): 60
Category:
Received: Apr. 24, 2023
Accepted: --
Published Online: May. 14, 2024
The Author Email:
CSTR:32186.14.