Automatic Stones Classification through a CNN-Based Approach

Sensors (Basel). 2022 Aug 21;22(16):6292. doi: 10.3390/s22166292.

Abstract

This paper presents an automatic recognition system for classifying stones belonging to different Calabrian quarries (Southern Italy). The tool for stone recognition has been developed in the SILPI project (acronym of "Sistema per l'Identificazione di Lapidei Per Immagini"), financed by POR Calabria FESR-FSE 2014-2020. Our study is based on the Convolutional Neural Network (CNNs) that is used in literature for many different tasks such as speech recognition, neural language processing, bioinformatics, image classification and much more. In particular, we propose a two-stage hybrid approach based on the use of a model of Deep Learning (DL), in our case the CNN, in the first stage and a model of Machine Learning (ML) in the second one. In this work, we discuss a possible solution to stones classification which uses a CNN for the feature extraction phase and the Softmax or Multinomial Logistic Regression (MLR), Support Vector Machine (SVM), k-Nearest Neighbors (kNN), Random Forest (RF) and Gaussian Naive Bayes (GNB) ML techniques in order to perform the classification phase basing our study on the approach called Transfer Learning (TL). We show the image acquisition process in order to collect adequate information for creating an opportune database of the stone typologies present in the Calabrian quarries, also performing the identification of quarries in the considered region. Finally, we show a comparison of different DL and ML combinations in our Two-Stage Hybrid Model solution.

Keywords: Convolutional Neural Network (CNN); Deep Learning (DL); Gaussian Naive Bayes (GNB); Machine Learning (ML); Random Forest (RF); Softmax; Support Vector Machine (SVM); Two-Stage Hybrid Model; k-Nearest Neighbors (kNN).

MeSH terms

  • Bayes Theorem
  • Cluster Analysis
  • Machine Learning
  • Neural Networks, Computer*
  • Support Vector Machine*

Grants and funding

This work was funded by the Calabria Region by “POR Calabria FESR-FSE 2014-2020, finanziamento di Progetti di Ricerca e Sviluppo, obiettivo specifico 1.2—Rafforzamento del Sistema Innovativo Regionale e Nazionale, Azione 1.2.2—Supporto alla realizzazione di Progetti Complessi di Attività di Ricerca e Sviluppo su poche Aree Tematiche di Rilievo e all’Applicazione di Soluzioni Tecnologiche Funzionali alla Realizzazione delle Strategie di S3”.