Laser Journal, Volume. 45, Issue 3, 100(2024)

Small target detection based on traffic signs

ZENG Tianhao* and CHEN Lin
Author Affiliations
  • [in Chinese]
  • show less

    Due to the low accuracy and false detection of small target detection algorithm in traffic sign detection , a new foreground fusion attention mechanism network called YOLO-Traffic is proposed. First , EIOU loss function is introduced to calculate the width of the predicted frame and the real frame respectively , and the dilated convolution is used to solve the problems existing in the original CIOU model. Secondly , the foreground attention mechanism F-ECA was added to fully extract the foreground information and suppress background noise. Finally , Kmeans+ + algorithm is used to replace the anchor frame obtained by Kmeans clustering to reallocate the corresponding feature layer and further improve the feature extraction ability. The experiment on TT100K traffic sign data set produced by Tsinghua University shows that compared with the original YOLOv5 network and the accuracy is increased by 2. 91% , the recall rate is in- creased by 2. 1% , the detection speed is 44 frames per second , and the final accuracy reaches 96. 89% . Hence , the proposed YOLO-Traffic network promotes the accuracy of traffic sign detection and model performance.

    Tools

    Get Citation

    Copy Citation Text

    ZENG Tianhao, CHEN Lin. Small target detection based on traffic signs[J]. Laser Journal, 2024, 45(3): 100

    Download Citation

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

    Category:

    Received: Jul. 21, 2023

    Accepted: --

    Published Online: Oct. 15, 2024

    The Author Email: Tianhao ZENG (273123948@qq.com)

    DOI:10.14016/j.cnki.jgzz.2024.03.100

    Topics