Intelligent Rock Recognition Based on Lightweight Network and Transfer Learning

Shun-yong LI , Qing-hui LI , Yu-man XING

Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (5) : 1774 -1782.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (5) : 1774-1782 DOI: 10.12404/j.issn.1671-1815.2400859
research-article
Intelligent Rock Recognition Based on Lightweight Network and Transfer Learning
Author information +
History +
PDF

Abstract

In rock image recognition, achieving rapid and accurate identification of rocks is crucial for the digitalization of rocks. Among the challenges faced in intelligent rock recognition is the issue of image blurring caused by environmental factors such as lighting and humidity. In light of this, a novel deep learning approach (MobileNetV3-small-RegNetX) was proposed for rock image recognition, which is suitable for scenarios with limited resources such as mobile devices. Building upon the RegNet network, transfer learning methods, combining the advantages of the MobileNetV3 residual structure with squeeze-and-excitation (SE) modules was employed to effectively optimize feature extraction and network structure, leading to a significant improvement in detection speed. To validate the accuracy of this approach, comparative experiments were conducted between the new model and current mainstream lightweight models (DenseNet and ShuffleNet). The results demonstrate that the new model proposed exhibits high precision (82.15%) and fast processing (0.06 GFLOPs). Additionally, the model demonstrates good adaptability to environmental factors such as lighting and humidity-induced image blurring.

Keywords

rock recognition / deep learning / image classification / transfer learning / MobileNet network

Cite this article

Download citation ▾
Shun-yong LI, Qing-hui LI, Yu-man XING. Intelligent Rock Recognition Based on Lightweight Network and Transfer Learning. Science Technology and Engineering, 2025, 25(5): 1774-1782 DOI:10.12404/j.issn.1671-1815.2400859

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (13117KB)

18

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉