15-494 S20 Cube Detector Project

Team:

Evan Druskin
SCS Undergrad
Kirubel Aklilu
SCS Undergrad

Cozmo has problems detecting cubes that are partially visible in his camera, and we sought out a way to improve his cube detection.
To solve this problem we constructed a neural network to do a binary classification cube or no cube.
The results turned out well, reporting 95-97% accuracy, but there are areas for improvement need.
Check out the powerpoint slides for a detailed description and there's a link to the code if you want to add any features.

Resources:

PowerPoint Slides: Code: Cozmo Tools: Cozmo SDK: Youtube Demos: