object-detector 1.9

In this Project I am Using YoloV4 for object detection. To detect object using YoloV4 and OpenCv you have to write many lines of code, but using this framework you can do this using few lines of code.

Installation

pip install object_detector

Extract Images from a video

Import ExtractImages class from object_detector Library

>>> from object_detector import ExtractImages

Create a object ext of ExtractImages class

>>> ext = ExtractImages(path=0)

path=0 if you want to extract images from a video, write video path in place of 0.

or you can create object specifing Output path

>>> ext = ExtractImages(path=0,op=MyOutput)

By default Output path is Output

Start Extracting Images.

>>> ext.extract("A","B","C")

A, B, C are the path where extracting images collected. Relative path of "A" is MyOutput/A

Note: We are extracting Images in multiple path such that we can run our object detection task simultaneously.

Object Detection

Import ObjectDetector class from object_detector Library

>>> from object_detector import ObjectDetector

Create a object obj of class ObjectDetector

>>> obj = ObjectDetector(weights="cars.weights", cfg="yolov4-custom.cfg", classes=['licence'])

Object detection is done using YoLo. cars.weights is the trained file generated by training image dataset by darknet, and we are using yolov4-custom.cfg.

>>> img = obj.detect_object(path)

Write path of the image, from which you want to detect object. img is numpy array of image. To the image from numpy array write

>>> import cv2
>>> cv2.imshow("Image_Name",img)

or You can write

>>> import matplotlib.pyplot as plt
>>> plt.imshow(img)

If you want image without label write the code as

>>> img = obj.detect_object(path,label=False)

If you want to img only when obj is detected from image write the code as

>>> img = detect_object(path,detected_only=True)

To read text on the image add following code

>>> text = extract_text()

Show or Write objects from a video or camera

Import LiveDetect class and make a object of this class.

l = LiveDetect(weights="cars.weights", cfg="yolov4-custom.cfg", classes=['licence'])

Save images when object detect on video.

l.write_from_video(path="Traffic.mp4")

Show images when object detect on video.

l.show_from_video(path="Traffic.mp4")

Save images when object detect on the dir.

l.write_from_dir(path="Traffic.mp4")

Show images when object detect on the dir.

l.show_from_dir(path="Traffic.mp4")