Using OpenCV to identify a dice roll
A small Python script that reads dice rolls out loud
By Quentin Golsteyn Python • Posted May 10, 2020
After a heated game of Yatzee! with my family, I began to suspect that the dice we were playing with were loaded. While all other players were able to roll fours, despite my best efforts, I was unable to roll this number. I lost the game, finishing last. Clearly something was up with these dice (or perhaps I am a sore loser).
This thread gave me all the info I needed to mathematically prove that the dice we were playing with were not fair. However, it required a lot of tedious work to get the data needed for this verification. If only I could automate that... And that's how I set out to teach my computer how to read dice rolls (and not actually verify if they were fair)!
For this project, I used OpenCV in Python to implement a small algorithm that counts the number of a dice roll. Other equipment needed: ✱a set of dice, a box to act as a dice roll container, and a smartphone, to provide a camera for this computer vision project.
The goal: processing a live video feed coming from my smartphone and overlay the number the computer saw for each rolled dice over the dice themselves.
This script assumes our camera is looking at our dice face down. I created ✱a testing rig to control for the camera position and ensure there was sufficient lighting.
This project requires OpenCV, sklearn and numpy to be installed on your machine.
pip install opencv-python numpy sklearn
I started by importing OpenCV and numpy, and by initializing a video feed.
import cv2import numpy as npfrom sklearn import cluster# Initialize a video feedcap = cv2.VideoCapture(0)
Everything happens in a loop, processing each video feed image one by one. I defined the general algorithm using a series of functions and I will go more in detail regarding the implementation of each function throughout this article.
Isolating dice dots
The first steps is to locate dice dots. BLOB stands for Binary Large OBject and refers to a group of connected pixels in a binary image.✱I applied the OpenCV simple blob detection algorithm.
Glare was a problem as the dice dots were glossy and reflecting the light from the smartphone. This created white spots which interfered with the detection algorithm. I applied a **median blur** on the frame before using the blob detection function provided by OpenCV.
Some blobs may not be associated with dice dots. I filtered them according to a inertia heuristic. I assumed the dice dots to be circular. Hence, they should appear circular in the image (as I am capturing them directly face down). A non-circular blob is either part of a dice face not directly face-up, or noise in the image.
Inertia is a ratio giving a measure of how round a blob is. A low inertia ratio indicates a more elongated blob (closer to a line), while a perfectly cirular blob will have an inertia ratio of one. Hence, I filtered contours that had an inertia ratio smaller than 0.6.
Identify individual dice
After isolating dice dots, I ended up with an array of dice dots contours. The next step is to merge these contours together into individual dice.
Density-based clustering was the method I used to perform this operation, specifically DBSCAN. I also considered k-means clustering, but this would have forced me to assume the number of dice I rolled each time (by having to set
k).✱Density-based clustering avoids having to set the number of clusters as it is automatically calculated based on the data itself. I calculated the centroid of each dot blob and used DBSCAN to assign a cluster to each dot, with hyper-parameter eps=40.
With these labels, I knew where each dice dot was in the image, and the dice it belonged to.
Reporting back our findings
Finally, I overlayed all the blobs and the numbers associated with each dice back on the image.
And because I thought it could be an interesting extension to this project, I made my computer say the numbers shown on the dice out loud.
There you have it! It was a good project to practice a computer vision problem. In case you need to have your computer automatically read dice numbers for you, you can find all the code for this small project in this Gist!
A front-end developer based in Vancouver, Canada.