Filter-Rick-And-Morty/rickFilter.py at main · mateobv07/Filter-Rick-And-Morty · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#Para este filtro nos basamos en el siguiente recurso:
#https://pysource.com/2019/03/25/pigs-nose-instagram-face-filter-opencv-with-python/
import cv2
import dlib
from rickCeja import createCeja
from rickHair import createHair
from rickBaba import createBaba
from rickTitulo import createTitulo
cap = cv2.VideoCapture(0)

detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")

while True:
    _, frame = cap.read()
    gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    faces = detector(frame)

    for face in faces:
        landmarks = predictor(gray_frame, face)
        #hair

        fromX, toX, fromY, toY, final = createHair(landmarks, frame)
        if fromX != 0:
            frame[fromY: toY , fromX: toX] = final
        #ceja
        fromX, toX, fromY, toY, final = createCeja(landmarks, frame)
        if fromX != 0:
            frame[fromY: toY , fromX: toX] = final
        #baba
        fromX, toX, fromY, toY, final = createBaba(landmarks, frame)
        if fromX != 0:
            frame[fromY: toY , fromX: toX] = final
        #titulo
        fromX, toX, fromY, toY, final = createTitulo(landmarks, frame)
        if fromX != 0:
            frame[fromY: toY , fromX: toX] = final
    cv2.imshow("RickMorty", frame)

    key = cv2.waitKey(1)
    if key == 27:
        break