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.
- Dataset download.
- Tools and Libraries.
- Code.
- References.
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.
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
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.
We have used keras.losses.CategoricalCrossentropy as our loss function, keras.optimizers.Adam as Optimizer and trained the model for 10 epochs.