Laser & Optoelectronics Progress, Volume. 58, Issue 14, 1410025(2021)

Road Surface Disease Detection Algorithm Based on Improved YOLOv4

Hui Luo, Chen Jia*, and Jian Li
Author Affiliations
  • School of Information Engineering, East China JiaoTong University, Nanchang, Jiangxi 330013, China
  • show less

    In order to solve the problems of multiple types of road surface diseases, large scale changes in the scale, and small sample data sets in road surface disease detection, a road surface multi-scale disease detecting algorithm based on improved YOLOv4 is proposed. First, the depth separable convolution method is used to replace the ordinary convolution method in the CSPDarknet-53 backbone network, which reduces the amount of network parameter calculations. Then, the loss function of YOLOv4 is improved based on the focal loss, which solves the problem of low detection accuracy caused by the imbalance of positive and negative samples in the process of network training. Finally, the YOLOv4 network is pre-trained with the help of transfer learning ideas, and the data set is expanded using methods such as flipping, cropping, brightness conversion, noise disturbance and other methods, so as to solve the problem of over-fitting of network training caused by insufficient samples of road surface disease. Experimental results show that, compared with original YOLOv4 detection network, the mean average precision of road surface disease detection based on YOLOv4+DC+FL algorithm can reach 93.64%, which increases by 3.25%. The detection time is 35.8 ms per picture, which is reduced by 7.9 ms.

    Tools

    Get Citation

    Copy Citation Text

    Hui Luo, Chen Jia, Jian Li. Road Surface Disease Detection Algorithm Based on Improved YOLOv4[J]. Laser & Optoelectronics Progress, 2021, 58(14): 1410025

    Download Citation

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

    Category: Image Processing

    Received: Aug. 17, 2020

    Accepted: Sep. 30, 2020

    Published Online: Jul. 6, 2021

    The Author Email: Jia Chen (jc_ecjtu@163.com)

    DOI:10.3788/LOP202158.1410025

    Topics