Next Article in Journal
Explicit Information Geometric Calculations of the Canonical Divergence of a Curve
Next Article in Special Issue
Optimizing the Neural Structure and Hyperparameters of Liquid State Machines Based on Evolutionary Membrane Algorithm
Previous Article in Journal
A Fractional-Order Compartmental Model of Vaccination for COVID-19 with the Fear Factor
Previous Article in Special Issue
A Modern Industrial Policy for the Czech Republic: Optimizing the Structure of Production
 
 
Article

Improving YOLOv4-Tiny’s Construction Machinery and Material Identification Method by Incorporating Attention Mechanism

by 1, 2,3, 1,4,* and 5
1
School of Electrical Electronic and Computer Science, Guangxi University of Science and Technology, Liuzhou 545006, China
2
School of Electrical Engineering, Yanshan University, Qinhuangdao 066004, China
3
Intelligent Technology Research, Institute of Global Research and Development Center, Guangxi LiuGong Machinery Company Limited, Liuzhou 545007, China
4
School of Resources and Environment, University of Electronic Science and Technology of China, Chengdu 611731, China
5
Guangxi Collaborative Innovation Centre for Earthmoving Machinery, Guangxi University of Science and Technology, Liuzhou 545006, China
*
Author to whom correspondence should be addressed.
Academic Editor: Stelios Papadakis
Mathematics 2022, 10(9), 1453; https://doi.org/10.3390/math10091453
Received: 16 March 2022 / Revised: 13 April 2022 / Accepted: 21 April 2022 / Published: 26 April 2022
(This article belongs to the Special Issue Advanced Aspects of Computational Intelligence with Its Applications)
To facilitate the development of intelligent unmanned loaders and improve the recognition accuracy of loaders in complex scenes, we propose a construction machinery and material target detection algorithm incorporating an attention mechanism (AM) to improve YOLOv4-Tiny. First, to ensure the robustness of the proposed algorithm, we adopt style migration and sliding window segmentation to increase the underlying dataset’s diversity. Second, to address the problem that YOLOv4-Tiny’s (the base network) framework only adopts a layer-by-layer connection form, which demonstrates an insufficient feature extraction ability, we adopt a multilayer cascaded residual module to deeply connect low- and high-level information. Finally, to filter redundant feature information and make the proposed algorithm focus more on important feature information, a channel AM is added to the base network to perform a secondary screening of feature information in the region of interest, which effectively improves the detection accuracy. In addition, to achieve small-scale object detection, a multiscale feature pyramid network structure is employed in the prediction module of the proposed algorithm to output two prediction networks with different scale sizes. The experimental results show that, compared with the traditional network structure, the proposed algorithm fully incorporates the advantages of residual networks and AM, which effectively improves its feature extraction ability and recognition accuracy of targets at different scales. The final proposed algorithm exhibits the features of high recognition accuracy and fast recognition speed, with mean average precision and detection speed reaching 96.82% and 134.4 fps, respectively. View Full-Text
Keywords: intelligent loader; style transfer; machine vision; CAM; YOLOv4-Tiny intelligent loader; style transfer; machine vision; CAM; YOLOv4-Tiny
Show Figures

Figure 1

MDPI and ACS Style

Yao, J.; Cai, D.; Fan, X.; Li, B. Improving YOLOv4-Tiny’s Construction Machinery and Material Identification Method by Incorporating Attention Mechanism. Mathematics 2022, 10, 1453. https://doi.org/10.3390/math10091453

AMA Style

Yao J, Cai D, Fan X, Li B. Improving YOLOv4-Tiny’s Construction Machinery and Material Identification Method by Incorporating Attention Mechanism. Mathematics. 2022; 10(9):1453. https://doi.org/10.3390/math10091453

Chicago/Turabian Style

Yao, Jiale, Dengsheng Cai, Xiangsuo Fan, and Bing Li. 2022. "Improving YOLOv4-Tiny’s Construction Machinery and Material Identification Method by Incorporating Attention Mechanism" Mathematics 10, no. 9: 1453. https://doi.org/10.3390/math10091453

Find Other Styles
Note that from the first issue of 2016, MDPI journals use article numbers instead of page numbers. See further details here.

Article Access Map by Country/Region

1
Back to TopTop