Open
Description
# Import SixDRepNet
from sixdrepnet import SixDRepNet
import cv2
# Create model
# Weights are automatically downloaded
model = SixDRepNet()
img = cv2.imread('/path/to/image.jpg')
pitch, yaw, roll = model.predict(img)
model.draw_axis(img, yaw, pitch, roll)
cv2.imshow("test_window", img)
cv2.waitKey(0)
Does this code measure the roll, pitch, and yaw of the entire image without a face detection model?
Will it be more accurate if I add a cropped image of the face to the img?
Metadata
Assignees
Labels
No labels