46 lines
999 B
Python
46 lines
999 B
Python
#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 m import *
|
|
mod.load_weights("model3-val-acc-0.74.model.keras")
|
|
|
|
c = cv.VideoCapture(0)
|
|
|
|
while True:
|
|
ret, frame = c.read()
|
|
|
|
pi = np.array(img(cv.cvtColor(frame, cv.COLOR_BGR2RGB)).resize((300, 300)).convert("RGB"))
|
|
#plt.imshow(pi)
|
|
#plt.show()
|
|
|
|
#print(pi.astype(float))
|
|
|
|
r = mod.predict(np.array([pi.astype(float)]))
|
|
|
|
cv.putText(frame, f"Y: {r[0][1]:.9f}", (10, 40), cv.FONT_HERSHEY_SIMPLEX, 1, (200,255,200), 2)
|
|
cv.putText(frame, f"N: {r[0][0]:.9f}", (10, 70), cv.FONT_HERSHEY_SIMPLEX, 1, (200,200,255), 2)
|
|
|
|
cv.imshow('me', frame)
|
|
|
|
if cv.waitKey(1) == ord('q'):
|
|
break
|
|
|
|
#print(frame)
|
|
|
|
c.release()
|
|
cv.destroyAllWindows()
|
|
|
|
#r = mod.predict(np.array([i]))
|
|
#print(r)
|