Electro-Optic Technology Application, Volume. 38, Issue 6, 60(2023)

Improve YOLOv5 Steel Surface Defect Detection Technology

LI Dengyue and JIANG Yueqiu
Author Affiliations
  • [in Chinese]
  • show less
    References(13)

    [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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Apr. 24, 2023

    Accepted: --

    Published Online: May. 14, 2024

    The Author Email:

    DOI:

    CSTR:32186.14.

    Topics