Industrial equipment detection algorithm under complex working conditions based on ROMS R-CNN

PLoS One. 2022 Apr 7;17(4):e0266444. doi: 10.1371/journal.pone.0266444. eCollection 2022.

Abstract

In the paper, we proposed a deep learning-based industrial equipment detection algorithm ROMS R-CNN (Rotation Occlusion Multi-Scale Region-CNN). It can solve the problem of inaccurate detection of industrial equipment under complex working conditions such as multi-scale ratio, rotation tilt, occlusion and overlap. The method proposed in this paper first is to construct the MobileNetV2 as the feature pyramid network, and then to combine high semantic information with high resolution information solved industrial equipment detection of different scales. Secondly, a specific rotation anchor scheme is proposed, and the data set is clustered through the k-means algorithm to obtain a specific aspect ratio. Combined with the rotation angle, a rotation anchor of any direction and size is generated to solve the problem of easy tilting of industrial equipment. Finally, a Non-Maximum Suppression algorithm with penalty factors is introduced to solve the overlapping in industrial equipment detection. The experimental results in common industrial equipment detection show that this method is better than other algorithms, significantly improves the missed detection and false detection, and the mAP reaches 0.939.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms*
  • Neural Networks, Computer*
  • Rotation
  • Semantics

Grants and funding

This work was financially supported by Science and technology development project of jilin province (NO.20200403075SF), Jilin Provincial Department of Science and Technology (NO. 20180201010GX), Jilin Provincial Department of Education (NO. JJKH20180440KJ). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.