6DRepNet icon indicating copy to clipboard operation
6DRepNet copied to clipboard

Where is the face detection model in test.py?

Open manadopeee opened this issue 11 months ago • 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?

manadopeee avatar Mar 04 '24 08:03 manadopeee