| giolekva | 550add7 | 2020-04-11 20:01:40 +0400 | [diff] [blame^] | 1 | from facenet_pytorch import MTCNN |
| 2 | import cv2 | ||||
| 3 | from PIL import Image, ImageDraw, ImageColor | ||||
| 4 | import numpy as np | ||||
| 5 | from matplotlib import pyplot as plt | ||||
| 6 | |||||
| 7 | mtcnn = MTCNN(keep_all=True) | ||||
| 8 | |||||
| 9 | img = Image.open("face.jpg") | ||||
| 10 | |||||
| 11 | boxes, _ = mtcnn.detect(img) | ||||
| 12 | draw = ImageDraw.Draw(img) | ||||
| 13 | for i, box in enumerate(boxes): | ||||
| 14 | draw.rectangle(((box[0], box[1]), (box[2], box[3])), outline="red") | ||||
| 15 | img.save("detected.jpg") | ||||
| 16 | #print(face) | ||||