| commit | edbaf32e7a3d7b25d17ec59bb0c236196e306f83 | [log] [tgz] |
|---|---|---|
| author | giolekva <giolekva@gmail.com> | Tue May 05 18:17:37 2020 +0400 |
| committer | giolekva <giolekva@gmail.com> | Tue May 05 18:17:37 2020 +0400 |
| tree | ec3686b206cdc9befbbe20166290ffee4fd11c19 | |
| parent | 8e060f8af01b3bcee957db66584eae6bcbb8483e [diff] [blame] |
face-detector: opencv based haarcascade
diff --git a/apps/face-detection/mtcnn.py b/apps/face-detection/mtcnn.py new file mode 100644 index 0000000..3a128a0 --- /dev/null +++ b/apps/face-detection/mtcnn.py
@@ -0,0 +1,11 @@ +from facenet_pytorch import MTCNN +from PIL import Image + + +def detect_faces(img_file): + mtcnn = MTCNN(keep_all=True) + ret = [] + with Image.open(img_file) as img: + for box in mtcnn.detect(img)[0]: + ret.append((box[0], box[1], box[2], box[3])) + return ret