Now Reading
Hands-On Guide To Detect Objects In Video In 5 Steps

Hands-On Guide To Detect Objects In Video In 5 Steps

Dr. Vaibhav Kumar
object detection in video
W3Schools

Object detection techniques are a trend nowadays in the field of computer vision. There are many techniques which are used to detect objects in the scene images and videos. Each of these techniques has its own advantages and limitations in terms of resources and execution time. Detecting objects in videos also requires a lot of technical knowledge and resources. 

So, there is always a search for an easy and quick method for object detection. In this article, we will show how to detect objects seen in the videos in only 5 steps. We will use the pixellib library in this task which detects objects using instance segmentation. We will also use the pre-trained Mask R-CNN model to identify the objects seen in the videos. In this implementation, we will detect vehicle objects in traffic video.

Instance Segmentation

Instance segmentation is a technique in computer vision that is used in object detection using image segmentation method. It identifies each instance of the objects present in the images or videos at the pixel level. In image segmentation, the visual input is splitted into segments to represent the object or part of objects by forming a collection of pixels. Instance segmentation identifies each instance of each object featured in the image instead of categorizing each pixel like in semantic segmentation.



Mask R-CNN

Mask R-CNN is a variant of Deep Neural Network proposed by Kaiming He et al at Facebook AI Research. This model is used to solve the object instance segmentation problem in computer vision. It detects objects in an image while simultaneously generating a high-quality segmentation mask for each instance. It is an extension of Faster R-CNN by adding a branch for predicting an object mask in parallel with the existing branch for bounding box recognition. The Mask R-CNN framework for instance segmentation is given below.

object detection in video

(Image Source: Original Research Paper)

Implementation

Now, we will discuss the steps through which we will detect the objects in a video.

1. Install library and dependencies

In the first step, we need to install the pixellib library and its dependencies.

! pip install pixellib
object detection in video

2. Load pre-trained Mask-RCNN weights

As we are going to use the Mas R-CNN model to detect objects, we will download its pre-trained weights.

!wget --quiet https://github.com/matterport/Mask_RCNN/releases/download/v2.0/mask_rcnn_coco.h5

3. Import libraries

Now, we will import the pixellib library that has been installed. We will also import the instance_segmentation class because we are going to detect objects using instance segmentation method.

import pixellib
from pixellib.instance import instance_segmentation

4. Instantiate instance segmentation model and load Mask-RCNN weights

In this step, we will instantiate the instance segmentation class that is provided by pixellib and load the Mask R-CNN model with its pre-trained weights.

segment_video = instance_segmentation()
segment_video.load_model("mask_rcnn_coco.h5")

See Also

5. Detect objects

In this step, we will process the object detection task by Mask R-CNN in a video. A random traffic video is used in which we want to detect vehicle objects.

Traffic Video

In this method, we set the frames per second that are the number of frames per second output video will have. 

segment_video.process_video("traffic_vid2.mp4", show_bboxes = True, frames_per_second= 15, output_video_name="object_detect.mp4")

























Finally, we will get the output video in the working directory. The time in this process depends on the lengths and size of the video. You should use a GPU for faster processing speed. For the above traffic video, we have got the following video in the out having detected objects.

You can define a function to get the videos from YouTube and pass them directly to the function above. So, using the above steps, we could discuss a very easy way to implement the task of object detection in videos. A person with very less knowledge of deep learning and computer vision can be able to detect objects using this way.

What Do You Think?

If you loved this story, do join our Telegram Community.


Also, you can write for us and be one of the 500+ experts who have contributed stories at AIM. Share your nominations here.

Copyright Analytics India Magazine Pvt Ltd

Scroll To Top