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")