Upload_Picture.py

import cv2
from PIL import Image, ImageTk
from tkinter import filedialog
from tkinter import *
# import matplotlib.pyplot as plt
import time
import numpy as np

"""
This module has 3 functions to upload the picture using different methods.
"""


def upload_from_local():
    """
    Function is called to upload file from local drive
    :return: A string. The path of the file uploaded.
    """
    filename = filedialog.askopenfilename(
        initialdir=r'C:\Users\admin\Pictures', title='Select a Image File'
    )
    return filename

def take_picture(filename):
    """
    Takes one picture from the default camera.
    :param filename: a string
    :return: Just the name of the file that is saved in working directory
            , The size of the image taken
    """
    cam = cv2.VideoCapture(0)
    cv2.namedWindow("Take an image hitting Enter, Hit Esc to go out")
    img_counter = 0

    while True:
        ret, frame = cam.read()
        if not ret:
            print("failed to grab frame")
            break

        cv2.imshow("Take an image hitting Enter, Hit Esc to go out", frame)

        k = cv2.waitKey(1)
        if k % 256 == 27:
            print("Escape hit, closing...")
            break
        if k % 256 == 13:
            # SPACE pressed
            img_name = "{}_{}.png".format(filename, img_counter)
            cv2.imwrite(img_name, frame)
            print("{} written!".format(img_name))
            img_counter += 1
            cam.release()
            cv2.destroyAllWindows()
            return img_name, ( np.array(frame).shape[1], np.array(frame).shape[0])


def take_picture_frames_in_video(filename):
    """
    Opens the camera and takes a video (does not record). Also Takes frames at discrete times and
    returns the frame.
    :param filename: a string
    :return: 2 objects. a generator which yields a camera frame as an array, and also the
    cv2.VideoCapture object
    """
    cam = cv2.VideoCapture(0)
    cv2.namedWindow("Take an image hitting Enter, Hit Esc to go out")
    img_counter = 0

    while True:
        ret, frame = cam.read()
        if not ret:
            print("failed to grab frame")
            break
        cv2.imshow("Take an image hitting Enter, Hit Esc to go out", frame)

        time.sleep(1 / 5)
        k = cv2.waitKey(1)
        if k % 256 == 27:
            print("Escape hit, closing...")
            img_name = "{}.png".format(filename)
            cv2.imwrite(img_name, frame)
            print("{} written!".format(img_name))
            break

        img_name = "{}_{}.png".format(filename, img_counter)
        print("{} array created!".format(img_name))
        img_counter += 1
        yield frame , cam

    cam.release()
    cv2.destroyAllWindows()


if __name__ == "__main__":
    upload_from_local()