Skip to content

Implementation of "Pothole Detection Using Location-Aware Convolutional Neural networks", keras with tensorflow backend. https://link.springer.com/article/10.1007/s13042-020-01078-7

Notifications You must be signed in to change notification settings

hanshenChen/pothole-detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pothole-detection

pothole detection -keras with tensorflow

Implementation from the paper "Pothole Detection Using Location-Aware Convolutional Neural networks"

Note:

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.

The structure of project

  • model_weights
  • preprocessing
  • models
  • utils
  • testwhole.py

Requirements

  • 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.

Usage

: The re-organized dataset are created by a simple preprocessing operation.

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.

: Testing

  1. 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;

About

Implementation of "Pothole Detection Using Location-Aware Convolutional Neural networks", keras with tensorflow backend. https://link.springer.com/article/10.1007/s13042-020-01078-7

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages