May-16-2021, 08:18 PM
(This post was last modified: May-18-2021, 04:52 PM by Yoriz.
Edit Reason: Added prefix
)
I have a program that opens video from the webcam and then performs functions such as saving an image and mirroring the live video. I am looking to add an overlay function where the user selects a PNG file that is overlayed on the webcam video. It works if I comment out the update function (line 32) in my code but then i have no webcam video displaying. Any help greatly appreciated. Thanks
from tkinter import *
import cv2
from PIL import Image, ImageTk
import time
from tkinter import filedialog
class App:
def __init__(self, video_source=0):
self.appName = "Kamera"
self.window = Tk()
self.window.title(self.appName)
self.window.resizable(0, 0)
# self.window.wm_iconbitmap("cam.ico")
self.window['bg'] = 'black'
self.video_source = video_source
self.vid = MyVideoCapture(self.video_source)
# self.label = Label(self.window, text=self.appName, font=15, bg='blue', fg='white').pack(side=TOP, fill=BOTH)
self.canvas = Canvas(self.window, width=self.vid.width, height=self.vid.height, bg='red')
self.canvas.pack()
self.btn_snapshot = Button(self.window, text="Snapshot", width=5, command=self.snapshot)
self.btn_snapshot.pack(side=LEFT, padx=10)
self.btn_flip = Button(self.window, text="Flip Image", width=7, command=self.flip_img)
self.btn_flip.pack(side=LEFT, padx=10)
self.btn_overlay = Button(self.window, text="Overlay", width=7, command=self.overlay)
self.btn_overlay.pack(side=LEFT, padx=10)
self.update()
self.window.mainloop()
def flip_img(self):
self.vid.flipped = not self.vid.flipped
def overlay(self):
file = filedialog.askopenfile(
mode='rb', defaultextension='.png',title="Choose Overlay Image", filetypes=[("PNG Files", '*.png')])
if file:
self.overlay_img = ImageTk.PhotoImage(file=file)
self.canvas.tag_raise(self.overlay_img)
self.canvas.create_image(0, 0, image=self.overlay_img, anchor=NW)
def snapshot(self):
check, frame = self.vid.getFrame()
if check:
filename = filedialog.asksaveasfilename(
defaultextension='.jpg', title="Choose Filename", filetypes=[("JPEG Files", '*.jpg')])
# image = "IMG-" + time.strftime("%H-%M-%S-%d-%m") + ".jpg"
cv2.imwrite(filename, cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
# msg = Label(self.window, text='Image Saved' + image, bg='black', fg='green').place(x=430, y=510)
else:
messagebox.showerror("paint says", "unable to save image ,\n something went wrong")
def update(self):
isTrue, frame = self.vid.getFrame()
if isTrue:
self.photo = ImageTk.PhotoImage(image=Image.fromarray(frame))
self.canvas.tag_lower(self.photo)
self.canvas.create_image(0, 0, image=self.photo, anchor=NW)
self.window.after(15, self.update)
class MyVideoCapture:
def __init__(self, video_source=0):
self.vid = cv2.VideoCapture(video_source)
if not self.vid.isOpened():
raise ValueError("Unable to open this camera \n select another video source", video_source)
self.width = self.vid.get(cv2.CAP_PROP_FRAME_WIDTH)
self.height = self.vid.get(cv2.CAP_PROP_FRAME_HEIGHT)
self.flipped = True
def getFrame(self):
if self.vid.isOpened():
isTrue, frame = self.vid.read()
if isTrue and self.flipped:
frame = cv2.flip(frame, 1)
if isTrue:
return (isTrue, cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
else:
return (isTrue, None)
else:
return (isTrue, None)
def __del__(self):
if self.vid.isOpened():
self.vid.release()
if __name__ == "__main__":
App()
