Я хочу показать точность прогноза и метки.Я выяснил, как это сделать для меток, но я не знаю, как показать точность для каждой прогнозируемой метки в окне захвата видео.
modelFaceRec = load_model('weights.h5')
listofLabelsNames = np.load('celebFaceList.npy')
video_capture = cv2.VideoCapture(1)
while True:
ret, frame = video_capture.read()
if ret:
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
faces = face_detector.detectMultiScale(gray, 1.3, 5)
for (x, y, w, h) in faces:
dataImg = load_Mydata(frame) # prepare the image for predict
dataImg = np.expand_dims(dataImg, 0)
v = modelFaceRec.predict(dataImg)
clasa = np.argmax(v)
cv2.rectangle(frame, (x, y), (x + w, y + h), (255, 0, 0), 2)
cv2.putText(frame, listofLabelsNames[clasa], (x, y), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0, 255, 0), 2)
Это функция load_data:
def load_Mydata(bgrImg):
averageImg = [129.1863, 104.7624, 93.5940]
cropFace = bgrImg.copy()
cropFace = cropFace[:, :, ::-1] # convert from RGB -> BGR
resizedCropFace = cv2.resize(cropFace, (224, 224), 0, 0, cv2.INTER_NEAREST)
resizedCropFace = resizedCropFace - averageImg
return resizedCropFace