/Rubiks-Cube-Face-Detection-Model

🧮 A deep learning object detection model built using SSD ResNet101 V1 FPN 640x640 🤖 in conjunction with TensorFlow object detection API. Trained on a custom dataset (300+ images) for 10 different labels, the model detects tile colors on Rubik's cube face with 98-100% accuracy 🎯 and only 0.2 total loss.

Primary LanguageJupyter Notebook

Watchers