Object Detection and Segmentation using Detectron2

Python application for real time object detection and segmentation.

Photo by Cory Schadt on Unsplash

Overview

In previous article I walked through with you on using YOLOv5 for real time object detection. In this article let’s develop a Python application for real time object detection using Detectron2, which is a Facebook AI Research’s next generation software system that implements state-of-the-art object detection algorithms.

YOLO using FastAPI WebSocket and React

The Application

The application uses opencv-python and Detectron2 pre-trained model to capture camera images and try to detect the objects.

• Git clone the repository.
• Run python setup.py install
• Install Detectron2 based on Python version and GPU availability
• Type d2 to start the application
• Type q to exit the application