Dmitri Villevald
Published © GPL3+

Transfer Learning with NVIDIA JetBot - Fun with Cones

Teach your robot to find a path in a maze of traffic cones using the camera and state-of-the-art deep learning model.

IntermediateFull instructions provided10 hours1,993

Things used in this project

Hardware components

NVIDIA JetBot
NVIDIA JetBot Wiki's Bill of Materials page lists everything you need to build JetBot, along with purchasing links from popular vendors.
×1
Computer with NVIDIA GPU
Needed to train the model
×1
BlueDot Trading 4” RC Racing Agility Cones, Orange – Set of 20
×20

Hand tools and fabrication machines

Soldering iron (generic)
Soldering iron (generic)

Story

Read more

Schematics

Main update loop

Explains the main update loop and shows how robot action depends on the model predictions and the robot's current state

Code

Three Jupyter notebooks for this project

The Jupyter notebooks data_collection_cones.ipynb, train_model_cones.ipynb and live_demo_cones.ipynb are located in subfolder traffic_cones_driving

Credits

Dmitri Villevald

Dmitri Villevald

5 projects • 12 followers
Manager, Advanced Analytics with a passion for Artificial Intelligence products.

Comments