114 lines
4.3 KiB
Python
114 lines
4.3 KiB
Python
import cv2
|
|
import numpy as np
|
|
import dlib
|
|
import math
|
|
|
|
cap=cv2.VideoCapture(0)
|
|
#cap=cv2.VideoCapture("debat.webm")
|
|
|
|
detector=dlib.get_frontal_face_detector()
|
|
predictor=dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")
|
|
|
|
def tr(c, o, coeff):
|
|
return(int((c-o)*coeff)+o)
|
|
|
|
def cube(image, pt1, pt2, a1, a2, a3):
|
|
color=(0, 255, 0)
|
|
epaisseur=2
|
|
offset=1.6
|
|
offset2=2
|
|
|
|
d_eyes=math.sqrt(math.pow(landmarks.part(36).x-landmarks.part(45).x, 2)+math.pow(landmarks.part(36).y-landmarks.part(45).y, 2))
|
|
|
|
ox1=int((-(pt2.y-pt1.y)+pt2.x-pt1.x)/2)+pt1.x
|
|
oy1=int(((pt2.x-pt1.x+pt2.y)-pt1.y)/2)+pt1.y
|
|
|
|
cv2.line(image,
|
|
(tr(pt1.x, ox1, offset), tr(pt1.y, oy1, offset)),
|
|
(tr(pt2.x, ox1, offset), tr(pt2.y, oy1, offset)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(pt2.x, ox1, offset), tr(pt2.y, oy1, offset)),
|
|
(tr(-(pt2.y-pt1.y)+pt2.x, ox1, offset), tr(pt2.x-pt1.x+pt2.y, oy1, offset)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(-(pt2.y-pt1.y)+pt2.x, ox1, offset), tr(pt2.x-pt1.x+pt2.y, oy1, offset)),
|
|
(tr(-(pt2.y-pt1.y)+pt1.x, ox1, offset), tr(pt2.x-pt1.x+pt1.y, oy1, offset)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(-(pt2.y-pt1.y)+pt1.x, ox1, offset), tr(pt2.x-pt1.x+pt1.y, oy1, offset)),
|
|
(tr(pt1.x, ox1, offset), tr(pt1.y, oy1, offset)),
|
|
color, epaisseur)
|
|
|
|
ox2=int((-(pt2.y-pt1.y)+pt2.x-pt1.x)/2)+pt1.x+int(a2)
|
|
oy2=int(((pt2.x-pt1.x+pt2.y)-pt1.y)/2)+pt1.y+int(a3)
|
|
|
|
cv2.line(image,
|
|
(tr(pt1.x+a2, ox2, offset2), tr(pt1.y+a3, oy2, offset2)),
|
|
(tr(pt2.x+a2, ox2, offset2), tr(pt2.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(pt2.x+a2, ox2, offset2), tr(pt2.y+a3, oy2, offset2)),
|
|
(tr(-(pt2.y-pt1.y)+pt2.x+a2, ox2, offset2), tr(pt2.x-pt1.x+pt2.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(-(pt2.y-pt1.y)+pt2.x+a2, ox2, offset2), tr(pt2.x-pt1.x+pt2.y+a3, oy2, offset2)),
|
|
(tr(-(pt2.y-pt1.y)+pt1.x+a2, ox2, offset2), tr(pt2.x-pt1.x+pt1.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(-(pt2.y-pt1.y)+pt1.x+a2, ox2, offset2), tr(pt2.x-pt1.x+pt1.y+a3, oy2, offset2)),
|
|
(tr(pt1.x+a2, ox2, offset2), tr(pt1.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
|
|
cv2.line(image,
|
|
(tr(pt1.x, ox1, offset), tr(pt1.y, oy1, offset)),
|
|
(tr(pt1.x+a2, ox2, offset2), tr(pt1.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(pt2.x, ox1, offset), tr(pt2.y, oy1, offset)),
|
|
(tr(pt2.x+a2, ox2, offset2), tr(pt2.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(-(pt2.y-pt1.y)+pt2.x, ox1, offset), tr(pt2.x-pt1.x+pt2.y, oy1, offset)),
|
|
(tr(-(pt2.y-pt1.y)+pt2.x+a2, ox2, offset2), tr(pt2.x-pt1.x+pt2.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
cv2.line(image,
|
|
(tr(-(pt2.y-pt1.y)+pt1.x, ox1, offset), tr(pt2.x-pt1.x+pt1.y, oy1, offset)),
|
|
(tr(-(pt2.y-pt1.y)+pt1.x+a2, ox2, offset2), tr(pt2.x-pt1.x+pt1.y+a3, oy2, offset2)),
|
|
color, epaisseur)
|
|
|
|
while True:
|
|
ret, frame=cap.read()
|
|
tickmark=cv2.getTickCount()
|
|
gray=cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
|
|
|
|
faces=detector(gray)
|
|
for face in faces:
|
|
x1=face.left()
|
|
y1=face.top()
|
|
x2=face.right()
|
|
y2=face.bottom()
|
|
|
|
landmarks=predictor(gray, face)
|
|
|
|
d_eyes=math.sqrt(math.pow(landmarks.part(36).x-landmarks.part(45).x, 2)+math.pow(landmarks.part(36).y-landmarks.part(45).y, 2))
|
|
d1=math.sqrt(math.pow(landmarks.part(36).x-landmarks.part(30).x, 2)+math.pow(landmarks.part(36).y-landmarks.part(30).y, 2))
|
|
d2=math.sqrt(math.pow(landmarks.part(45).x-landmarks.part(30).x, 2)+math.pow(landmarks.part(45).y-landmarks.part(30).y, 2))
|
|
coeff=d1+d2
|
|
|
|
a1=int(250*(landmarks.part(36).y-landmarks.part(45).y)/coeff)
|
|
a2=int(250*(d1-d2)/coeff)
|
|
cosb=min((math.pow(d2, 2)-math.pow(d1, 2)+math.pow(d_eyes, 2))/(2*d2*d_eyes), 1)
|
|
a3=int(250*(d2*math.sin(math.acos(cosb))-coeff/4)/coeff)
|
|
|
|
cube(frame, landmarks.part(36), landmarks.part(45), a1, a2, a3)
|
|
cv2.imshow("Frame", frame)
|
|
|
|
key=cv2.waitKey(1)&0xFF
|
|
if key==ord('q'):
|
|
break
|
|
|
|
cap.release()
|
|
cv2.destroyAllWindows()
|
|
1
|