Mineral Rock Classification Using Convolutional Neural Network

Abstract

The main aim of the research is to build a model that can effectively predict the type of mineral rocks. Rocks can be predicted by observing it is colour, shape and chemical composition. On-site technicians need to apply different techniques on rock sample in order to predict rock type. Technicians need to apply different techniques on rock samples, so it is a time-consuming process, and sometimes the predictions may be accurate, and sometimes predictions may be false. When predictions are false, it might show a negative impact in several ways for workers and organization as well. We considered an image dataset of rock types, namely Biotite, Bornite, Chrysocolla, Malachite, Muscovite, Pyrite, and Quartz. We applied CNN (Convolutional Neural Network) Algorithm to get a better prediction of different mineral rocks. Nowadays, CNN is mainly used for image classification and image recognition tasks.