Abstract
Image-based inspection is a growing area with a large scope of automation. The automatic classification of vehicle damages would make the insurance claim much faster and more efficient. This can effectively reduce the claiming cost. This paper presents, an image classification model using an adapted version of pre-trained convolutional neural networks. The pre-trained neural networks were, the VGG-19 and DenseNet-169. The proposed model is a pipeline that established with fully connected layers for additional damage classification. The final proposed model improves the feature extraction process. The dataset had a class imbalance problem, so a weighted loss function had been used to solve such problem. The model employed binary cross-entropy as a loss function, and sigmoid activation was applied to the output layers as independent layers. Finally, the model presents a multi-label classifier, where one image may be assigned to many labels. The model classifies vehicle damage through five classes: broken glass, broken headlights, broken taillights, scratches, and dents. A four-layer neural network was employed for the classification, along with several regularization approaches to handle overfitting problem. The final results showed that the DenseNet-169 had a better accuracy of 81%, whereas VGG-19 had a 78%. Another approach had been proposed where it had a mix of transfer and ensemble learning approaches. This final approach had an accuracy of 85.5% and F1-scores of 0.855.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
More From: Journal of Ambient Intelligence and Humanized Computing
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.