Pet project to extract rubiks cube color using OpenCV. Working demo in youtube.
main.py- This is the main file that reads video frames and detect cube faces. The following operations are performed in this file:- Opening the video frame and gray scaling it.
- Edges are detected after gaussian blurring with canny edge detector.
- For clear visibiity of edges, dilution and erosion are performed.
- Bounding circles over 9 individual square pieces in the current face of the cube is detected and drawn.
- After capturing 9 small squares in a face, a small cube is drawn at top of the output video frame upon pressing letter
e. - Press letter
qto quit the video frame.
hsv_color_range- This file contains hsv color range for each of the six colors in the cube. It also has function to detect color of a region based on its mean hsv values.utilities- This file holds all other additional utilities for manipulation.
- Python 3.10 used.
- Main dependencies are listed here.
- Enable conda/other python environment and install the dependencies
- Show one face of the cube at a time in front of the camera.
- Press
eto capture and draw its face colors. - After capturing 6 faces, press
eagain to recapture from beginning. - Press
qto quit. - For more clarity, watch this demo.
- Use the captured 6 faces to generate a 3D model of cube.
- Un-availability of code for Andrej's similar work.