Lubricating Grease Thickness Classification of Steel Wire Rope Surface Based on GEMR-MobileViT

Abstract

Proper surface lubrication with optimal grease thickness is essential for extending steel wire rope service life. To achieve automated lubrication quality control and address challenges like variable lighting and motion blur that degrade recognition accuracy in practical settings, this paper proposes an improved lightweight GEMR-MobileViT. The model is designed to identify the grease thickness on steel wire rope surfaces while mitigating the high parameters and computational complexity of existing models. In this model, part of the standard convolution is replaced by GhostConv, a novel efficient multi-scale attention (EMA) module is introduced into the local expression part of the MobileViT block, and the structure of residual connections within the MobileViT block is designed. A transfer learning method is then employed. A custom dataset of steel wire rope lubrication images was constructed for model training. The experimental results demonstrated that GEMR-MobileViT achieved a recognition accuracy of 96.63% across five grease thickness categories, with 4.19 M params and 1.31 GFLOPs computational complexity. Compared to the pre-improvement version, recognition accuracy improved by 4.4%, while its parameters and computational complexity were reduced by 15.2% and 10.3%, respectively. When compared with current mainstream classification models such as ConvNeXtV2, EfficientNetV2, EdgeNeXt, NextViT, and MobileNetV4, our GEMR-MobileViT achieved superior recognition accuracy and demonstrated significant advantages in its model parameters, striking a good balance between recognition precision and model size. The proposed model facilitates deployment in steel wire rope lubrication working sites, enabling the real-time monitoring of surface grease thickness, thereby offering a novel approach for automating steel wire rope maintenance.

Keywords

• image classification
• deep learning
• lightweight model
• MobileViT
• EMA
• steel wire rope