learnopencv/Mask-RCNN

Deep learning based Object Detection and Instance Segmentation using Mask RCNN in OpenCV (Python / C++)

This repository contains code for Deep learning based Object Detection and Instance Segmentation using Mask RCNN in OpenCV (Python / C++) blog post.

Python

wget http://download.tensorflow.org/models/object_detection/mask_rcnn_inception_v2_coco_2018_01_28.tar.gz tar zxvf mask_rcnn_inception_v2_coco_2018_01_28.tar.gz

Download and extract the needed model files.

Usage Examples :

Python

python3 mask_rcnn.py --image=cars.jpg python3 mask_rcnn.py --video=cars.mp4

It starts the webcam - if no argument provided.

C++

Compile using:

g++ -ggdb `pkg-config --cflags --libs /Users/snayak/opencv/build/unix-install/opencv.pc` mask_rcnn.cpp -o mask_rcnn.out

Run using: ./mask_rcnn.out --image=cars.jpg ./mask_rcnn.out --video=cars.mp4

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.