Как преобразовать модель распознавания лиц в файл h5 - PullRequest
Новая
       48

Как преобразовать модель распознавания лиц в файл h5

0 голосов
/ 15 апреля 2020

Я пытаюсь создать приложение для распознавания лиц во флаттере, где оно будет выделять самое большое лицо среди множества лиц и сохранять его. Не могли бы вы помочь, как преобразовать файл в файл h5 / tflite

Вот мои коды: 

import cv2
import sys

#Face detection image
imagePath = sys.argv[1]
image = cv2.imread(imagePath)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

#Face dectection 
faceCascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
faces = faceCascade.detectMultiScale(
    gray,
    scaleFactor=1.3,
    minNeighbors=3,
    minSize=(30, 30)
)
print(len(faces))
print("[INFO] Found {0} Faces!".format(len(faces)))
a=0
for (x, y, w, h) in faces:

    cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)     
    a1 = h*w

    if(a1>a):
        a = a1
        h1 = h
        w1 = w
        roi_color = image[y:y + h1, x:x + w1]

print("[INFO] Object found. Saving locally.") 
image1 = roi_color
cv2.imwrite('faces.jpg', image1)
...