Skip to content

This is a project based on ConvNets used to identify whether a road is clean or dirty. We have used MobileNet as our base architecture and the weights are based on imagenet.

FaizalKarim280280/Clean-Dirty-Road-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

37 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PROJECT TITLE: CLEAN/DIRTY ROAD DETECTION USING TRANSFER LEARNING

Description:

This is a project based on ConvNets used to identify whether a road is clean or dirty. We have used MobileNet as our base architecture and the weights are based on imagenet.

Alt text

Contents

  1. Dataset download.
  2. Tools and Libraries.
  3. Code.
  4. References.

1. Dataset download

We have used web scraping techniques to download 238 images of both clean and dirty roads. This dataset was completely used for training the model.

2. Tools and Libraries

Alt text

In this project we will be using the Keras Deep Learning Library and we will be running it on top of the Tensorflow backend.
Others include:

  • Numpy
  • Matplotlib
  • OpenCV

3. Code

ClassifierModel.py is the main code which encapsulates the CNN architecture that was used for this project.

  • We have used keras.applications.mobilenet_v2 to import the MobileNetV2 architecture for our base model, with training = False.
  • On top of it, we have added two dense layers consisting of 512 and 256 units both having ReLU activation followed by Dropout.
  • The output layer is a simple 2 unit softmax layer.

Training Accuracy and Loss

loss

We have used keras.losses.CategoricalCrossentropy as our loss function, keras.optimizers.Adam as Optimizer and trained the model for 10 epochs.

4. References

About

This is a project based on ConvNets used to identify whether a road is clean or dirty. We have used MobileNet as our base architecture and the weights are based on imagenet.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages