The "Where's Waldo? AI" project uses machine learning techniques to detect Waldo in images. This project leverages the YOLOv5 object detection model to identify Waldo in various images, making it easier to locate him in the classic "Where's Waldo?" puzzles. The motivation behind this project was to put AI into something unorthodox at first and see the possible use cases of it.
- Object Detection: Uses YOLOv5 to detect Waldo in images.
- Image Annotation: Visualizes detection results with bounding boxes around Waldo.
- Python 3.7 or later
torch
1.7.0 or latertorchvision
0.8.0 or later
- Download the Dataset from here: https://universe.roboflow.com/tan-rmi/whereiswaldo-ssmol/dataset/1
- Download the weights from here: https://drive.google.com/drive/folders/1frst8S5lJDwAnfBQNfSjsnZaJSoTjUTX?usp=sharing
- Load the IPYNB file and follow the steps
- Test the model yourself by uploading an image you want ( Steps mentioned in the IPYNB file)
A few results from training the model during development phase :