DEV Community

Cover image for Building a Real-Time Object Detection Application with YOLO
Abhinav Anand
Abhinav Anand

Posted on

Building a Real-Time Object Detection Application with YOLO

Object detection has become one of the most exciting applications of artificial intelligence, enabling machines to understand and interpret visual data. In this tutorial, we will walk through the steps to create a real-time object detection application using the YOLO (You Only Look Once) algorithm. This powerful model allows for fast and accurate detection of objects in images and videos, making it suitable for various applications, from surveillance to autonomous vehicles.

Table of Contents

  1. What is Object Detection?
  2. Understanding YOLO
  3. Setting Up Your Environment
  4. Installing Dependencies
  5. Building the Object Detection App
  6. Potential Use Cases
  7. Conclusion

What is Object Detection?

Object detection is a computer vision task that involves identifying and locating objects within an image or video stream. Unlike image classification, which only determines what objects are present, object detection provides bounding boxes around the detected objects, along with their class labels.

Understanding YOLO

YOLO, which stands for "You Only Look Once," is a state-of-the-art, real-time object detection algorithm. The primary advantage of YOLO is its speed; it processes images in real-time while maintaining high accuracy. YOLO divides the input image into a grid and predicts bounding boxes and probabilities for each grid cell, allowing it to detect multiple objects in a single pass.

Setting Up Your Environment

Before we dive into the code, make sure you have the following installed:

  • Python 3.x: Download from python.org.
  • OpenCV: A library for computer vision tasks.
  • NumPy: A library for numerical computations.
  • TensorFlow or PyTorch: Depending on your preference for running the YOLO model.

Creating a Virtual Environment (Optional)

Creating a virtual environment can help manage dependencies effectively:

python -m venv yolovenv
source yolovenv/bin/activate  # On Windows use yolovenv\Scripts\activate
Enter fullscreen mode Exit fullscreen mode

Installing Dependencies

Install the required libraries using pip:

pip install opencv-python numpy
Enter fullscreen mode Exit fullscreen mode

For YOLO, you may need to download the pre-trained weights and configuration files. You can find YOLOv3 weights and config on the official YOLO website.

Building the Object Detection App

Now, let’s create a Python script that will use YOLO for real-time object detection.

Step 1: Load YOLO

Create a new Python file named object_detection.py and start by importing the necessary libraries and loading the YOLO model:

import cv2
import numpy as np

# Load YOLO
net = cv2.dnn.readNet("yolov3.weights", "yolov3.cfg")
layer_names = net.getLayerNames()
output_layers = [layer_names[i[0] - 1] for i in net.getUnconnectedOutLayers()]
Enter fullscreen mode Exit fullscreen mode

Step 2: Process the Video Stream

Next, we’ll capture video from the webcam and process each frame to detect objects:

# Capture video from webcam
cap = cv2.VideoCapture(0)

while True:
    ret, frame = cap.read()
    height, width, channels = frame.shape

    # Prepare the image for YOLO
    blob = cv2.dnn.blobFromImage(frame, 0.00392, (416, 416), (0, 0, 0), True, crop=False)
    net.setInput(blob)
    outs = net.forward(output_layers)

    # Process the detections
    class_ids = []
    confidences = []
    boxes = []

    for out in outs:
        for detection in out:
            scores = detection[5:]
            class_id = np.argmax(scores)
            confidence = scores[class_id]
            if confidence > 0.5:  # Adjust confidence threshold as needed
                # Object detected
                center_x = int(detection[0] * width)
                center_y = int(detection[1] * height)
                w = int(detection[2] * width)
                h = int(detection[3] * height)

                # Rectangle coordinates
                x = int(center_x - w / 2)
                y = int(center_y - h / 2)

                boxes.append([x, y, w, h])
                confidences.append(float(confidence))
                class_ids.append(class_id)

    # Apply Non-Max Suppression
    indexes = cv2.dnn.NMSBoxes(boxes, confidences, 0.5, 0.4)

    # Draw bounding boxes and labels on the frame
    for i in range(len(boxes)):
        if i in indexes:
            x, y, w, h = boxes[i]
            label = str(classes[class_ids[i]])
            cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
            cv2.putText(frame, label, (x, y + 30), cv2.FONT_HERSHEY_PLAIN, 3, (0, 255, 0), 3)

    cv2.imshow("Image", frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()
Enter fullscreen mode Exit fullscreen mode

Step 3: Running the Application

To run the application, execute the script:

python object_detection.py
Enter fullscreen mode Exit fullscreen mode

You should see a window displaying the webcam feed with detected objects highlighted in real time.

Potential Use Cases

Real-time object detection has a wide array of applications, including:

  • Surveillance Systems: Automatically detecting intruders or unusual activities in security footage.
  • Autonomous Vehicles: Identifying pedestrians, traffic signs, and other vehicles for navigation.
  • Retail Analytics: Analyzing customer behavior and traffic patterns in stores.
  • Augmented Reality: Enhancing user experiences by detecting and interacting with real-world objects.

Conclusion

Congratulations! You’ve successfully built a real-time object detection application using YOLO. This powerful algorithm opens up numerous possibilities for applications across various fields. As you explore further, consider diving into more advanced topics, such as fine-tuning YOLO for specific object detection tasks or integrating this application with other systems.

If you're interested in pursuing a career in AI and want to learn how to become a successful AI engineer, check out this Roadmap To Become Successful AI Engineer for a detailed roadmap.

Feel free to share your thoughts, questions, or experiences in the comments below. Happy coding!


Top comments (0)