pothole detection -keras with tensorflow
Implementation from the paper "Pothole Detection Using Location-Aware Convolutional Neural networks"
The public experimental code has been changed a lot of times in doing the experiment,which will cause a lot of redundant issues, it was not designed in accordance with the priciples and norms of software engineering.
- model_weights
- preprocessing
- models
- utils
- testwhole.py
- Tensorflow
- Keras
- Opencv3
- Pothole dataset:
The dataset(each image is 800 × 600 in JPG format) was first released in the Data Science Hackathon, a computer vision challenge sponsored by IBM,the Machine Intelligence Institute of Africa, and Cortex Logic, which took place in Johannesburg in September 2017.
you can download the dataset fromTraining & Test Data and Python Notebooks
The orginal images(each image is 3680 × 2760 in JPG format) was first released in paper:
[1] S. Nienaber, M.J. Booysen, R.S. Kroon, “Detecting potholes using simple image processing
techniques and real-world footage”, SATC, July 2015, Pretoria, South Africa.
[2] S. Nienaber, R.S. Kroon, M.J. Booysen , “A Comparison of Low-Cost Monocular Vision Techniques
for Pothole Distance Estimation”, IEEE CIVTS, December 2015, Cape Town, South Africa.
1、because the files name of two dataset is not the same,
we have to find the same picture and rename it at the same with the help of some Similarity Image Finder software;
2、run preprocessing/b3org_to_roi.py to crop the road images from the origianl image and create the new labels;
3、hold out 800 training images as a validation set;
4、the road images are resized to 352*244 to meet the LCNN model requirement.
5、run preprocessing/s6create_heatmap.py to create the ground truth of heatmap;
6、run preprocessing/create_patch.py to create the patch dataset.
- download the model's weights patchnet & lcnn
2、you can change the path of dataset and other parameters in testwhole.py;
3、run testwhole.py;