Images_from_Video.py

import datetime
import os

import cv2
from DuplicateRemover import DuplicateRemover


class ImagesFromVideo:
    """
    Extract the Images from video and
    apply pre-processing algorithms
    on images
    """

    def __init__(self, logger, f_name=""):
        self.filename = f_name
        """
        Creating an object
        """
        self.logger = logger

    def readfile(self):
        try:
            """
            Read the video from specified path
            """
            cam = cv2.VideoCapture(str(self.filename))
            return cam
        except Exception as e:
            self.logger.error("File Not Found at this specific directory...")

    def detect_blur_images(self, image):
        """
        detect the blur image
        """
        threshold = 60
        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        fm = cv2.Laplacian(gray, cv2.CV_64F).var()
        text = "Blurry"
        if fm >= threshold:
            text = "Not Blurry"
            data = [text, image]
            return data

        data = [text, image]
        return data

    def get_frames_times(self,cam):
        video_frames = cam.get(cv2.CAP_PROP_FRAME_COUNT)
        fps = int(cam.get(cv2.CAP_PROP_FPS))

        # calculate duration of the video
        seconds = int(video_frames / fps)
        video_time = str(datetime.timedelta(seconds=seconds))
        return {
            "video_time": video_time,
            "no_of_frames": video_frames,
            "frames_per_second": fps
        }

    def processing(self, cam):
        currentframe = 0
        try:
            while True:
                """
                reading from frames
                """
                # cam.set(cv2.CAP_PROP_POS_MSEC, (currentframe * 1000))

                ret, frame = cam.read()
                if ret:
                    """
                    set the name of frame
                    """
                    try:
                        name = os.path.dirname(os.path.abspath(__file__)) + '/' +\
                               'frames/frame' + str(
                            currentframe) + '.jpg'

                        """
                        writing the extracted images
                        """
                        data = self.detect_blur_images(frame)
                        cv2.imwrite(name, data[1])
                        # if data[0] == "Not Blurry":
                        #     cv2.imwrite(name, data[1])

                        """
                        increasing counter so that it will
                        show how many frames are created
                        """
                        currentframe += 1
                    except Exception as e:
                        self.logger.error(e)
                else:
                    break
        except Exception as e:
            self.logger.error(e)

    def detect_duplicate_images(self):
        """
        Remove Duplicates Images
        """
        dr = DuplicateRemover(os.path.dirname(os.path.abspath(__file__)) + '/' + "frames/", self.logger)
        dr.find_duplicates()