|
2 years ago | |
---|---|---|
images | 2 years ago | |
.gitignore | 2 years ago | |
BoggleCV.py | 2 years ago | |
BoggleCVPipeline.py | 2 years ago | |
LICENSE | 3 years ago | |
README.md | 2 years ago | |
SortImages.py | 2 years ago | |
TensorflowBoggle.py | 2 years ago | |
TestBoggleCVPipeline.py | 2 years ago | |
bogglecv-demo.mp4 | 2 years ago | |
full-pipeline.ipynb | 2 years ago | |
io-json-formats.txt | 3 years ago | |
minimal-pipeline.ipynb | 2 years ago | |
tensorflow-boggle.ipynb | 2 years ago |
README.md
BoggleCV
Using OpenCV and Tensorflow to identify a Boggle board in an image and decode what letters are on the board.
Try it!
BoggleCV is deployed on my online boggle site, try it here! You can take a picture of a boggle board with your phone, or upload an image.
Demo Video
Explanation
Here are the various stages of the computer vision and machine learning pipeline, with a sample image.
- An HSV threshold is used to find all the blue pixels, and create a mask.
- A contour finding algorithm from OpenCV finds the outline of large blue blob in the image.
Image with Box Drawn | Mask to Find Blue Pixels |
---|---|
![]() |
![]() |
- The scipy signal processing library is used on the points in the outline to find corners
- The box is cut out from the original image and warped to become a square.
Corner Finding | Warped Image |
---|---|
![]() |
![]() |
- Next, the individual letters must be found. A sum of all the pixels in each row is taken, and the dips in brightness are found.
- The same is repeated for the columns.
- The positions of the rows and columns are used to cut out each letter from the image, and the letters are resized to 30x30 for easier use in machine learning.
Row/Column Sum | Letters Cut Out |
---|---|
![]() |
![]() |
- This entire process is repeated over many images (and videos) to create a dataset of thousands of boggle letter images.
- The images are labelled by hand. For videos, this is much easier since you only have to label it once for the whole video.
- A Tensorflow model is trained on the dataset and exported to a file. It uses a convolutional neural network (CNN) which is optimal for images and other 2-dimensional data. After the CNN, a few 1-dimensional layers are used to get the output to an easier format.
#convolutional (2-dimensional) layers
model = models.Sequential()
model.add(layers.Conv2D(64, (3, 3), activation='relu', input_shape=(IMG_DIM, IMG_DIM, 1)))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(32, (3, 3), activation='relu'))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(16, (3, 3), activation='relu'))
#data is flattened from 2 dimensions to 1
model.add(layers.Flatten())
#a fully-connected layer is added for more flexibility in combining features
model.add(layers.Dense(100, activation='relu'))
#the final layer has 1 output for each letter of the alphabet, and they are expressed as percentages that it thinks it might be of each letter. usually, one letter is close to 100% and the rest are close to 0%, which means it is very sure of its decision. But if it is less sure of a result, the number would be lower, e.g. 80%, and it might have 1 or 2 other options that were 15% and 5% for example. This information can be used, possibly to ask the user for a better quality image, or a confirmation that the letter is correct.
model.add(layers.Dense(26, activation="softmax"))
Note: steps 8-10 only have to be done one time to set up the pipeline.
- Finally, the trained model is used to identify the letters in a new image. (See the output below.)
Right now, the model has about 88% accuracy on images it has never seen before. This is a pretty good start, but I am planning to improve it by deploying this to my site where people can play boggle, and create a system for uploading images and correcting any mistakes that the model makes. Then I will take this new data and re-train the model to achieve higher accuracy.