Skip to content
Branch: master
Find file History
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_R-CNN_demo.ipynb update demo notebook with keypoints (#479) Feb 22, 2019
README.md 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
predictor.py
shapes_dataset_demo.ipynb Fix typo for shapes_dataset_demo.ipynb (#885) Jun 12, 2019
shapes_pruning.ipynb
webcam.py Initial release Oct 24, 2018

README.md

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 webcam.py --min-image-size 800
# can also run it on the CPU
python webcam.py --min-image-size 300 MODEL.DEVICE cpu
# or change the model that you want to use
python webcam.py --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 webcam.py --min-image-size 300 --show-mask-heatmaps MODEL.DEVICE cpu

With Docker

Build the image with the tag maskrcnn-benchmark (check INSTALL.md 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 \
    -e DISPLAY=${DISPLAY} \
    --privileged \
    -v /tmp/.X11-unix:/tmp/.X11-unix \
    --device=/dev/video0:/dev/video0 \
    --ipc=host maskrcnn-benchmark \
    python demo/webcam.py --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.