Varun221 / rubiks_camsolver

A Rubik's cube Solver that takes cube colors through webcam and prints out the solution moves
14 stars 2 forks source link

Model Weights #29

Open Keval-P opened 1 year ago

Keval-P commented 1 year ago

Where Can I find the Model weights because the links are not valid at current moment. So can you please let us know where can we find the model weights and also the dataset.

Varun221 commented 1 year ago

Hey, I created this project a long time ago and due to a hard drive problem I lost the dataset and weights. I can tell you how I created it. 1) You can create a simple dataset by taking images of a Rubik's cube in different orientations, backgrounds, with/without colors around it, etc. (Preferably around 70-80 images for a good detector) 2) Label the images using LabelImg software. (You can find more directions here) You can also use the dataset provided in the above blog, but it won't generalize for all types of Rubik's cubes. 3) The training code should be the same.

Do write if you face any problems. Happy to help!

Keval-P commented 1 year ago

Just needed one more help for the solving the cube like in which Way we should scan the cube and give the input string to the kociema algorithm.

Varun221 commented 1 year ago

You can see the function to convert to string here. The input to the function is created in detect_face.py

Keval-P commented 1 year ago

Can you please tell me what is format of the output for the detect_face.py and what is string format for the kociemba algorithm which you use for solving the algorithm.