#from PIL import Image
from sys import argv, exit
import numpy as np
import cv2 as cv

from tensorflow.keras.utils import array_to_img as img

from matplotlib import pyplot as plt

if len(argv) != 2:
    exit(1)

#i = np.array(Image.open(argv[1]).convert('RGB').resize((300,300)))

from test2 import *
m.load_weights("model3.keras")

c = cv.VideoCapture(argv[1])

frames = []

while True:
    ret, frame = c.read()

    if not ret:
        break

    r = cv.resize(frame, dsize=(300, 300), interpolation=cv.INTER_CUBIC)
    r = cv.cvtColor(r, cv.COLOR_BGR2RGB)
    frames.append(r)

c.release()

frames = np.array(frames)
res = m.predict(frames, batch_size = 64)

plt.plot([i[0] for i in res])
plt.show()