frame_grabbin.py

import cv2
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation


def grab_frame(cap):
    _, frame = cap.read()
    return cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)


def update(i):
    im.set_data(grab_frame(cap))


def close(event):
    if event.key == 'q':
        plt.close(event.canvas.figure)


cap = cv2.VideoCapture(0)
fig, ax = plt.subplots()
im = ax.imshow(grab_frame(cap))
ani = FuncAnimation(plt.gcf(), update, interval=20)
cid = plt.gcf().canvas.mpl_connect("key_press_event", close)
plt.show()