thohemp/6DRepNet

Where is the face detection model in test.py?

manadopeee opened this issue · 0 comments

# 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?