Skip to content

Code for training the SSD Mobilenet V2 COCO object detection model to detect the Alphabot2 robot in the lab space.

License

Notifications You must be signed in to change notification settings

Sarmyt/masters_object_detection

Repository files navigation

Instructions

Training Link: https://colab.research.google.com/drive/1_qgZ8TM8YvMEtIhbZXg-49j5NzODt6hU?usp=sharing

The code found at this link is used for training the SSD Mobilenet V2 COCO object detection model for localizing the Alphabot2 robot in the work space. Make sure to enable GPU in the Google Colab Environment for fast training.

Training configuration parameters can be found in the "ssd_mobilenet_v2_coco.config" file.

Training images can be found in the "images" folder and the respective manually created annotations for these images can be found in the "annotations" folder.

About

Code for training the SSD Mobilenet V2 COCO object detection model to detect the Alphabot2 robot in the lab space.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published