Я хочу показать точность и метку прогнозируемой фотографии в окне видеозахвата.Как я могу это сделать? - PullRequest
0 голосов
/ 22 июня 2019

Я хочу показать точность прогноза и метки.Я выяснил, как это сделать для меток, но я не знаю, как показать точность для каждой прогнозируемой метки в окне захвата видео.

    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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...