Marble Classification Using Deep Neural Networks

Abstract

Deep learning, which has been described as the processing and interpretation of data, is now widely used. In this study, deep neural networks are used for the classification of marbles which can be used in the industry. For this purpose most used marbles images were obtained from companies in Turkey and 28-class dataset was created. Then VGG16, ResNet and LeNet models were trained on this dataset. Data augmentation was performed to have class balance. To evaluate the models performance accuracy metric is used. In the VGG16 model, fine tunning was applied and %97 accuracy was achieved. In experimental studies, models were trained with different parameter settings. The performances of the models are given comparatively. The fact that both new dataset and deep neural networks are used for the first time in marble classification are among the positive aspects of this study. It is planned to integrate the models produced in the future studies into mobile based expert systems.