Efficient and Lightweight Framework for Real-Time Ore Image Segmentation Based on Deep Learning

Minerals

Guodong Sun¹  Delong Huang¹  Yang Zhang¹  Junjie Jia¹  Chenyun Xiong¹  Le Cheng¹ 

[1] School of Mechanical Engineering, Hubei University of Technology, Wuhan 430068, China;

关键词: real-time;    deep learning;    ore images;    instance segmentation;   

DOI  :  10.3390/min12050526

来源: DOAJ

【 摘 要 】

Image segmentation approaches have been utilized to determine the particle size distribution of crushed ores in the past decades. It is not possible to deploy large and high-powered computing equipment due to the complex working environment, so existing algorithms are difficult to apply in practical engineering. This article presents a novel efficient and lightweight framework for ore image segmentation to discern full and independent ores. First, a lightweight backbone is introduced for feature extraction while reducing computational complexity. Then, we propose a compact pyramid network to process the data obtained from the backbone to reduce unnecessary semantic information and computation. Finally, an optimized detection head is proposed to obtain the feature to maintain accuracy. Extensive experimental results demonstrate the effectiveness of our method, which achieves 40 frames per second on our new ore image dataset with a very small model size. Meanwhile, our method maintains a high level of accuracy—67.68% in $A{P}_{50}^{box}$ and 46.73% in $A{P}_{50}^{mask}$—compared with state-of-the-art approaches.

【 授权许可】

Unknown