Home > Archive > 2019 > Volume 9 Number 2 (Apr. 2019) >
IJMLC 2019 Vol.9(2): 201-207 ISSN: 2010-3700
DOI: 10.18178/ijmlc.2019.9.2.787

Evaluating on the Transfer Learning of CNN Architectures to a Construction Material Image Classification Task

Supaporn Bunrit, Nittaya Kerdprasop, and Kittisak Kerdprasop

Abstract—Various sub-tasks on modern construction management system require automatic or semi-automatic processes in handling the operation inside. Especially for construction progress monitoring task, the automatic process in classifying the difference of each construction material from an image is necessary in the preliminary stage. The more the preciseness in automatic classifying, the more the exactness in assessment of each material had been used. Subsequently, the progress of the construction can be evaluated with the highest degree of reliability. As a result, classification of construction material images is very essential process for automatic progress monitoring. Whereas, the similarities in material image appearances are the major classifying challenges. All most all existing related works have been studied based on hand-designed features of which the classified accuracy still not much appreciated from different studied datasets. In our work, automatic feature extracted method from the prominent technique in deep learning, convolution neural network (CNN), is proposed. The pre-trained CNN architectures of AlexNet and GoogleNet are adopt with the task of construction material images classification in the concept of transfer learning. Both of fixed feature extractor and fine-tuning schemes of transfer learning are technically implemented and evaluated. Analyzing results from the two pre-trained architectures expose very impressive and interesting circumstances to the studied dataset. Entirely, fine-tuning scheme of GoogleNet reveals the highest classification result by 95.50 percent of accuracy.

Index Terms—Convolution neural network (CNN), deep learning, transfer learning, construction material, image classification.

The authors are with the School of Computer Engineering, SUT, Thailand (corresponding author: S. Bunrit; Tel.: +66944961244; e-mail: sbunrit@sut.ac.th, nittaya@sut.ac.th, kerdpras@sut.ac.th).


Cite: Supaporn Bunrit, Nittaya Kerdprasop, and Kittisak Kerdprasop, "Evaluating on the Transfer Learning of CNN Architectures to a Construction Material Image Classification Task," International Journal of Machine Learning and Computing vol. 9, no. 2, pp. 201-207, 2019.

General Information

  • ISSN: 2010-3700 (Online)
  • Abbreviated Title: Int. J. Mach. Learn. Comput.
  • Frequency: Bimonthly
  • DOI: 10.18178/IJMLC
  • Editor-in-Chief: Dr. Lin Huang
  • Executive Editor:  Ms. Cherry L. Chen
  • Abstracing/Indexing: Inspec (IET), Google Scholar, Crossref, ProQuest, Electronic Journals Library.
  • E-mail: ijmlc@ejournal.net

Article Metrics