Skip to content

Object detection and classification model based on CNN using YOLOv3 along with some image enhancement techniques.

Notifications You must be signed in to change notification settings

gouravbarkle/Object-classification-and-detection-model-using-YOLOv3

Repository files navigation

Object-classification-and-detection-model-using-YOLOv3

1. Data preprocessing

The data preprocessing phase mainly includes the methodology based on the intensity transformation technique specifically into the spatial domain to enhance the image quality before feeding into the model.

Adjusting the Brightness and Contrast of the image

The parameters α>0 and β are often called the gain and bias parameters; sometimes these parameters are said to control contrast and brightness respectively. g(i,j)=α⋅f(i,j)+β f(x): source image pixels and g(x) :output image pixels

original image

alt text

Image after tuning Brightness and Contrast

alt text

Histogram Equalization

An image histogram gives a graphical representation of the distribution of pixel intensities in a digital image. The x-axis indicates the range of values the variable can take. This range can be divided into a series of intervals called bins. The y-axis shows the count of how many values fall within that interval.

histogram representation of original image

alt text

histogram representation after Equalization

alt text

Output after Histogram Equalization

alt text

2. Object classification and detection using YOLO

  • What is Yolo?

    • YOLO (You Only Look Once) object detection algorithm, which is one of the most effective object detection algorithms.
    • It takes the entire image in a single instance and predicts the bounding box coordinates and class probabilities for these boxes.
    • YOLO takes input and make predictions on it in single pass only, it uses CNN in the background for task completion.
  • YOLO Architecture

    • 53 Conv layers called Darknet-53 stacked with 53 more layers producing 106 layers in total.
    • No pooling layer used here instead of pooling it uses convolution layer with a filter 1X1 to down sample the input, as a result it enables & helps to improve ability to detect small object.
    • YOLO V3, detects object at 3 different scales and at the layer 82,94 and 106.
    • It uses three different stride values as 32,16 and 8 for detecting different types of features from the input.

Final sample input output

Sample input 1

alt text

Sample input 2

alt text

Sample output 1

alt text

Sample output 2

alt text

About

Object detection and classification model based on CNN using YOLOv3 along with some image enhancement techniques.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published