Preprint Article Version 1 Preserved in Portico This version is not peer-reviewed

Autonomous and Real-time Rock Image Classification using Convolutional Neural Networks

Version 1 : Received: 14 September 2021 / Approved: 16 September 2021 / Online: 16 September 2021 (13:38:55 CEST)

How to cite: Pascual, A.D.; McIsaac, K.; Osinski, G. Autonomous and Real-time Rock Image Classification using Convolutional Neural Networks. Preprints 2021, 2021090285. https://doi.org/10.20944/preprints202109.0285.v1 Pascual, A.D.; McIsaac, K.; Osinski, G. Autonomous and Real-time Rock Image Classification using Convolutional Neural Networks. Preprints 2021, 2021090285. https://doi.org/10.20944/preprints202109.0285.v1

Abstract

Autonomous image recognition has numerous potential applications in the field of planetary science and geology. For instance, having the ability to classify images of rocks would allow geologists to have immediate feedback without having to bring back samples to the laboratory. Also, planetary rovers could classify rocks in remote places and even in other planets without needing human intervention. Shu et al. classified 9 different types of rock images using a Support Vector Machine (SVM) with the image features extracted autonomously. Through this method, the authors achieved a test accuracy of 96.71%. In this research, Convolutional Neural Networks(CNN) have been used to classify the same set of rock images. Results show that a 3-layer network obtains an average accuracy of 99.60% across 10 trials on the test set. A version of Self-taught Learning was also implemented to prove the generalizability of the features extracted by the CNN. Finally, one model has been chosen to be deployed on a mobile device to demonstrate practicality and portability. The deployed model achieves a perfect classification accuracy on the test set, while taking only 0.068 seconds to make a prediction, equivalent to about 14 frames per second.

Keywords

remote sensing; deep learning; image classification

Subject

Computer Science and Mathematics, Artificial Intelligence and Machine Learning

Comments (0)

We encourage comments and feedback from a broad range of readers. See criteria for comments and our Diversity statement.

Leave a public comment
Send a private comment to the author(s)
* All users must log in before leaving a comment
Views 0
Downloads 0
Comments 0
Metrics 0


×
Alerts
Notify me about updates to this article or when a peer-reviewed version is published.
We use cookies on our website to ensure you get the best experience.
Read more about our cookies here.