Skip to content
Using MaskRCNN to solve Where's waldo
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
Mask_RCNN @ cbff80f
data
models
results
.gitignore
.gitmodules
LICENSE
README.md
data.py
predict.py
train.py
waldo_config.py

README.md

Deepwaldo

Using Mask-RCNN to solve Where's Waldo. Refer this for detailed description of the problem and the solution.

Some results

Data

Training and validation data sets used for this project are put in data folder. For custom datasets, use via-via tool to annotate.

Running

You can either set your configurations in the waldo_config.py file or use the default. Then use the following commands for training and predicting.

Training

This will download the Mask-RCNN model trained on coco dataset to the MODEL_DIR folder and trains a model with the data in the DATA_DIR folder.

python train.py

Detecting

Shows a popup with the waldo detected in the image.

python predict.py [MODEL PATH] [PATH TO IMAGE]
# for example
python predict.py models/logs/waldo20180612T1628/mask_rcnn_waldo_0030.h5 data/val/5.jpg
You can’t perform that action at this time.