Research on small scale fireworks target detection in complex environment
DOI:
Author:
Affiliation:

1.School of Automation,Nanjing Institute of technology,Nanjing,Jiangsu;2.Province Research Center of Intellisense Technology and System,Nanjing,Jiangsu

Clc Number:

TP391

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Aiming at the problem of low efficiency and accuracy of fireworks detection due to the small size of fire target and the confusion of fire feature with actual scene in complex environment, a small scale fireworks target detection method based on improved YOLOv5 is proposed. Firstly, a fourth detection layer is added to the third detection layer output in the original YOLOv5 model, so as to obtain a larger feature map for small target detection and strengthen the feature extraction capability of the network model. Secondly, in order to solve the problem that the target is prone to miss detection in the shielded scene, GIOU_Loss used to calculate the regression loss function of the target frame in the original network is replaced by DIOU_Loss. Finally, TensorRT is used to compress and accelerate the optimization of the model, and it is deployed to the Jetson TX2 development board for accelerated reasoning experiments. More fireworks scene data are constructed by replication enhancement method and a large number of experimental results show that the proposed method not only has a fast convergence speed, but also has a higher accuracy for small scale fire spot detection. It is suitable for popularization and application.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:July 10,2022
  • Revised:July 25,2022
  • Adopted:August 16,2022
  • Online:
  • Published:

Address:No. 219, Ningliu Road, Nanjing, Jiangsu Province

Postcode:210044

Phone:025-58731025