YoloKerasFaceDetection
YoloKerasFaceDetection copied to clipboard
how can I get boxes when run yolov2_tiny-face.h5
Hello, I'm a newbie I want to gets boxes from yolov2_tiny-face.h5. Here is my code
from keras.models import load_model
from PIL import Image, ImageDraw
import numpy as np
input_image = Image.open('test.jpg')
image = input_image.resize((416, 416))
image = np.array(image)
image = image / 255.
image = np.expand_dims(image, 0)
model = load_model('yolov2_tiny-face.h5')
model.summary()
outputs = model.predict(image)
I got outputs.shape = (1, 13, 13, 30), same as last layer's shape, seem not correct Could you help me please!
Hey, could you solve this? I have the same issue.