Improving classification of road surface conditions via road area extraction and contrastive learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Trinh, Linh, Anwar, Ali, Mercelis, Siegfried
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910535329513472
author Trinh, Linh
Anwar, Ali
Mercelis, Siegfried
author_facet Trinh, Linh
Anwar, Ali
Mercelis, Siegfried
contents Maintaining roads is crucial to economic growth and citizen well-being because roads are a vital means of transportation. In various countries, the inspection of road surfaces is still done manually, however, to automate it, research interest is now focused on detecting the road surface defects via the visual data. While, previous research has been focused on deep learning methods which tend to process the entire image and leads to heavy computational cost. In this study, we focus our attention on improving the classification performance while keeping the computational cost of our solution low. Instead of processing the whole image, we introduce a segmentation model to only focus the downstream classification model to the road surface in the image. Furthermore, we employ contrastive learning during model training to improve the road surface condition classification. Our experiments on the public RTK dataset demonstrate a significant improvement in our proposed method when compared to previous works.
format Preprint
id arxiv_https___arxiv_org_abs_2407_14418
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Improving classification of road surface conditions via road area extraction and contrastive learning
Trinh, Linh
Anwar, Ali
Mercelis, Siegfried
Computer Vision and Pattern Recognition
Maintaining roads is crucial to economic growth and citizen well-being because roads are a vital means of transportation. In various countries, the inspection of road surfaces is still done manually, however, to automate it, research interest is now focused on detecting the road surface defects via the visual data. While, previous research has been focused on deep learning methods which tend to process the entire image and leads to heavy computational cost. In this study, we focus our attention on improving the classification performance while keeping the computational cost of our solution low. Instead of processing the whole image, we introduce a segmentation model to only focus the downstream classification model to the road surface in the image. Furthermore, we employ contrastive learning during model training to improve the road surface condition classification. Our experiments on the public RTK dataset demonstrate a significant improvement in our proposed method when compared to previous works.
title Improving classification of road surface conditions via road area extraction and contrastive learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.14418