Skip to content
Branch: master
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.
Mask_R-CNN_demo.ipynb update demo notebook with keypoints (#479) Feb 22, 2019 add the option to use a `FORCE_CUDA` to force cuda installation on do… Mar 31, 2019
demo_e2e_mask_rcnn_R_50_FPN_1x.png Initial release Oct 24, 2018
demo_e2e_mask_rcnn_X_101_32x8d_FPN_1x.png Initial release Oct 24, 2018
panoptic_segmentation_shapes_dataset_demo.ipynb Panoptic Segmentation (#928) Jun 25, 2019
shapes_dataset_demo.ipynb Fix typo for shapes_dataset_demo.ipynb (#885) Jun 12, 2019
shapes_pruning.ipynb Initial release Oct 24, 2018

Webcam and Jupyter notebook demo

This folder contains a simple webcam demo that illustrates how you can use maskrcnn_benchmark for inference.

With your preferred environment

You can start it by running it from this folder, using one of the following commands:

# by default, it runs on the GPU
# for best results, use min-image-size 800
python --min-image-size 800
# can also run it on the CPU
python --min-image-size 300 MODEL.DEVICE cpu
# or change the model that you want to use
python --config-file ../configs/caffe2/e2e_mask_rcnn_R_101_FPN_1x_caffe2.yaml --min-image-size 300 MODEL.DEVICE cpu
# in order to see the probability heatmaps, pass --show-mask-heatmaps
python --min-image-size 300 --show-mask-heatmaps MODEL.DEVICE cpu

With Docker

Build the image with the tag maskrcnn-benchmark (check for instructions)

Adjust permissions of the X server host (be careful with this step, refer to here for alternatives)

xhost +

Then run a container with the demo:

docker run --rm -it \
    --privileged \
    -v /tmp/.X11-unix:/tmp/.X11-unix \
    --device=/dev/video0:/dev/video0 \
    --ipc=host maskrcnn-benchmark \
    python demo/ --min-image-size 300 \
    --config-file configs/caffe2/e2e_mask_rcnn_R_50_FPN_1x_caffe2.yaml

DISCLAIMER: This was tested for an Ubuntu 16.04 machine, the volume mapping may vary depending on your platform

You can’t perform that action at this time.