mirror of
https://github.com/deepinsight/insightface.git
synced 2025-12-30 16:12:27 +00:00
24 lines
699 B
Python
24 lines
699 B
Python
import cv2
|
|
import numpy as np
|
|
import os
|
|
import insightface
|
|
from insightface.app import FaceAnalysis
|
|
from insightface.data import get_image as ins_get_image
|
|
|
|
if __name__ == '__main__':
|
|
app = FaceAnalysis(allowed_modules=['detection', 'landmark_2d_106'])
|
|
app.prepare(ctx_id=0, det_size=(640, 640))
|
|
img = ins_get_image('t1')
|
|
faces = app.get(img)
|
|
#assert len(faces)==6
|
|
tim = img.copy()
|
|
color = (200, 160, 75)
|
|
for face in faces:
|
|
lmk = face.landmark_2d_106
|
|
lmk = np.round(lmk).astype(np.int)
|
|
for i in range(lmk.shape[0]):
|
|
p = tuple(lmk[i])
|
|
cv2.circle(tim, p, 1, color, 1, cv2.LINE_AA)
|
|
cv2.imwrite('./test_out.jpg', tim)
|
|
|