DEV Community

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

Posted on

190 2 3 3 4

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!


Image of AssemblyAI tool

Challenge Submission: SpeechCraft - AI-Powered Speech Analysis for Better Communication

SpeechCraft is an advanced real-time speech analytics platform that transforms spoken words into actionable insights. Using cutting-edge AI technology from AssemblyAI, it provides instant transcription while analyzing multiple dimensions of speech performance.

Read full post

Top comments (8)

Collapse
 
nmnir profile image
Nahum

YOLO current version is 11. Why are you using v3?

Collapse
 
abhinowww profile image
Abhinav Anand

I will soon upload the article with YOLO v11 also.

Collapse
 
john12 profile image
john

Nice project !!

Collapse
 
abhinowww profile image
Abhinav Anand

thank you

Collapse
 
matikiri profile image
Matikiri

The best

Collapse
 
eruvierda profile image
Ahmad Firdaus

is there any dedicated data model for plant especially tropical plant in yOLO? and which YOLO version should I use for plant recognition and measurement

Collapse
 
moisebis00 profile image
Moïse Bisimwa • Edited

WOW, very powerful
Is there a best model than face-api.js for js, if so, I will appreciate your help thank you

Collapse
 
abhinowww profile image
Abhinav Anand

Thank You

Some comments may only be visible to logged-in visitors. Sign in to view all comments.

A Workflow Copilot. Tailored to You.

Pieces.app image

Our desktop app, with its intelligent copilot, streamlines coding by generating snippets, extracting code from screenshots, and accelerating problem-solving.

Read the docs

👋 Kindness is contagious

Explore a sea of insights with this enlightening post, highly esteemed within the nurturing DEV Community. Coders of all stripes are invited to participate and contribute to our shared knowledge.

Expressing gratitude with a simple "thank you" can make a big impact. Leave your thanks in the comments!

On DEV, exchanging ideas smooths our way and strengthens our community bonds. Found this useful? A quick note of thanks to the author can mean a lot.

Okay