Mine underground object detection algorithm based on TTFNet and anchor-free

Abstract

To solve the problem of poor object detection effect caused by uneven light and high noise in underground mines, this study proposes a TTFNet (training-time-friendly network)-based object detection algorithm for underground mines. First, CenterNet and TTFNet algorithms are introduced, then pooling is introduced into CSPNet basic structure to design a lightweight feature extraction network, at the same time optimizing the feature fusion way in the original algorithm, optimizing residual shrinkage network structure, and introducing it into object detection task. Experiments were conducted on the established underground data set. The results show that compared with the original algorithm, our proposed algorithm can still maintain similar accuracy while significantly reducing model parameters; compared with other anchor-based detection algorithms, it has achieved similar overall performance.

Keywords

• anchor-free
• underground object detection
• convolutional neural network
• lightweight model