Road Bump Detection -- using a Deep Learning Model

Detecting Bumps on Road Surfaces using an object detection model trained using the TF Object Detection API and inferenced on Jetson Nano Using TensorRT.

Check the output of the object detection model here:

https://www.youtube.com/watch?v=kqT52hL7pLM&feature=youtu.be

Files Overview

Training_Colab_MobileNetV2.ipynb

• This is where we trained the model on COLAB using the TensorFlow object detection API.

TRT_Optimize.ipynb

• This is where we convert the frozen TensorFlow graph to a TensorRT accelerated model.

TRT_Inference_ZED.ipynb

• This is where we use the model during the inference operation (final deployment).
• This file uses the helper_functions.py file