Проблема с графиком matplotlib
Написал по проекту прогу, которая должна считывать пульс лица по видеокамере(в настоящем времени), с помощью opencv и numpy. Проблема заключается в том что для того чтобы вывести мат. формулу, обязательную по проекту, необходимо отследить переменную heartbeat_values на графике, а я не знаю как это правильно сделать. При попытке просто сравнить со временем (time.time()) выдаёт след. ошибку: Сообщение = x and y must have same first dimension, but have shapes (1,) and (128,). Прикладываю код ниже:
import numpy as np
from matplotlib import pyplot as plt
import cv2
import io
import time
cap = cv2.VideoCapture(0)
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1280)
cap.set(cv2.CAP_PROP_FPS, 30)
x, y, w, h = 800, 500, 100, 100
heartbeat_count = 128
heartbeat_values = [0]*heartbeat_count
heartbeat_times = [time.time()]*heartbeat_count
fig = plt.figure()
ax = fig.add_subplot(111)
while(True):
ret, frame = cap.read()
img = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
crop_img = img[y:y + h, x:x + w]
heartbeat_values = heartbeat_values[1:] + [np.average(crop_img)]
heartbeat_times = [time.time()]
ax.plot(heartbeat_times, heartbeat_values)
fig.canvas.draw()
plot_img_np = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
plot_img_np = plot_img_np.reshape(fig.canvas.get_width_height()[::-1] + (3,))
plt.cla()
cv2.imshow('Crop', crop_img)
cv2.imshow('Graph', plot_img_np)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()