Abstract
Traditional ore identification methods usually analyze the physical or chemical properties of ore samples and get conclusions, but those methods are not efficient when faced with large quantities of samples. In this paper, the YOLOv4 object detection algorithm is introduced and trained as a classifier of 7 common ores, which can make classification and position prediction for the image of different kinds of ore samples. The model has a high recognition accuracy, and the loss function is easy to converge. Moreover, the model also adapted several data augmentation methods and utilize Auto Multi-Scale Retinex with Color Restoration (MSRCR) for reducing the influence photograph environment of the ore specimen. The paper also uses the random forest algorithm to enhance the adaptability of the model in complex environments. The model has the advantages of low cost, short detection time, and strong generalization ability, covering most application scenarios from raw ore to ore products.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Cutmore, N.G., Liu, Y., Middleton, A.G.: Ore characterization and sorting. Miner. Eng. 10(4), 421–426 (1997)
Xiao, D., Le, B.T., Ha, T.T.L.: Iron ore identification method using reflectance spectrometer and a deep neural network framework. Spectrochemica Acta Part A: Mol. Biomol. Spectrosc. 248, 119168 (2021)
Donskoi, E., et al.: Utilization of optical image analysis and automatic texture classification for iron ore particle characterization. Miner. Eng. 20(5), 461–471 (2007)
Laine, S., Lappalainen, H., Jämsä-Jounela, S.L.: On-line determination of ore type using cluster analysis and neural networks. Miner. Eng. 8(6), 637–648 (1995)
Singh, V., Singh, T.N., Singh, V.: Image processing applications for customized mining and ore classification. Arab. J. Geosci. 4(7), 1163–1171 (2011)
Kumar, P.A., Chatterjee, S., Gorai, A.K.: Development of machine vision-based ore classification model using support vector machine (SVM) algorithm. Arab. J. Geosci. 10(5), 107 (2017)
Naresh, S., et al.: Textural identification of basaltic rock mass using image processing and neural network. Comput. Geosci. 14(2), 301–310 (2010)
Veerendra, S., Rao, S.M.: Application of image processing and radial basis neural network techniques for ore sorting and ore classification. Miner. Eng. 18(15), 1412–1420 (2005)
Lee, C.H., Lin, C.W.: A two-phase fashion apparel detection method based on YOLOv4. Appl. Sci. 11(9), 3782 (2021)
Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: unified, real-time object detection. In: Proceedings of the 2016 IEEE Computer Society Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA, 27–30 June 2016, pp. 779–788 (2016)
Bochkovskiy, A., Wang, C.Y., Liao, H.Y.M.: YOLOv4: Optimal Speed and Accuracy of Object Detection. arXiv 2020, arXiv:2004.10934 (2020)
Mahesh, P.: Random forest classifier for remote sensing classification. Int. J. Remote Sens. 26(1), 217–222 (2005)
Minerals Identification Dataset [EB/OL]. https://www.kaggle.com/asiedubrempong/minerals-identification-dataset (2019)
Ghiasi, G., Lin, T.Y., Le, Q.V.: Dropblock: a regularization method for convolutional networks. arXiv preprint arXiv:1810.12890 (2018)
Zheng, Z.H., et al.: Enhancing geometric factors in model learning and inference for object detection and instance segmentation. arXiv preprint arXiv:2005.03572 (2020)
Wang, J., Perez, L.: The effectiveness of data augmentation in image classification using deep learning. Convolutional Neural Netw. Vis. Recognit arXiv:1712.01621 (2017)
Kisantal, M., Wojna, Z., Murawski, J., Naruniec, J., Cho, K.: Augmentation for small object detection. arXiv preprint arXiv:1902.07296 (2019)
Wang, C.Y., Bochkovskiy, A., Liao, H.Y.M.: Scaled-YOLOv4: scaling cross stage partial network. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 13029–13038 (2021)
Li, H., Deng, L., Yang, C., Liu, J., Gu, Z.: Enhanced YOLOv3 tiny network for real-time ship detection from visual image. IEEE Access 9, 16692–16706 (2021)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Wang, T. (2022). Ore Detection Method Based on YOLOv4. In: Jain, L.C., Kountchev, R., Tai, Y., Kountcheva, R. (eds) 3D Imaging—Multidimensional Signal Processing and Deep Learning. Smart Innovation, Systems and Technologies, vol 297. Springer, Singapore. https://doi.org/10.1007/978-981-19-2448-4_24
Download citation
DOI: https://doi.org/10.1007/978-981-19-2448-4_24
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-19-2447-7
Online ISBN: 978-981-19-2448-4
eBook Packages: Intelligent Technologies and RoboticsIntelligent Technologies and Robotics (R0)