Skip to content

zl-gan/image-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Classification of Road Features Using Transfer Learning and CNN

Introduction

The study proposes a classification approach that utilizes transfer learning and convolutional neural networks (CNNs) to address the road feature classification problem. The dataset contained 7616 images of roundabout, crosswalk, overpass, and intersection from the MLRSNet dataset and manually extracted satellite images from Malaysia using Google Earth Pro. Transfer learning models such as ResNet50, MobileNetV2, VGG19 and InceptionV3 were also explored for road feature classification.

Methodology

The dataset used in this study is primarily sourced from the MLRSNet dataset and additional satellite images of roundabout, intersection, crosswalk, and overpass in Malaysia were obtained by taking screenshots of those features using Google Earth Pro software. A custom CNN model is created in this study to perform classification of roundabout, crosswalk, intersection, and overpass. Transfer learning models such as ResNet50, MobileNetV2, VGG19 and InceptionV3 are used to compare the accuracy with the CNN model developed.

Results

The best performing model for road feature classification are ResNet50 and VGG-19 with an accuracy of 98.7132%.

Conclusion

The study demonstrates the effectiveness of transfer learning and CNNs for the classification of road features. The proposed approach offers a practical and efficient solution for accurate identification and categorization of road features, contributing to the advancement of intelligent transportation systems and enhancing overall road safety and efficiency.

About

This project is part of an assignment for the course CDS521 Multimodal Information Retrieval provided by School of Computer Sciences, USM as part of their Masters of Science in Data Science and Analytics program.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors