mraber/face-detect-mqtt
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add check for image

Browse Source
This commit is contained in:
se1exin 2022-05-13 20:09:37 +10:00
parent debef7771a
commit 1673e591f7
No known key found for this signature in database
GPG Key ID: 8C5633201B25C57D
2 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
README.md Normal file
View File

@ -0,0 +1,7 @@
Install on Raspberry Pi
https://github.com/superuser789/MediaPipe-on-RaspberryPi/issues/10#issuecomment-1079703581
sudo apt install python3-opencv

4
tracker.py
View File

@ -43,14 +43,18 @@ class Tracker(object):
self.img = img self.img = img
def detect_face(self): def detect_face(self):
if self.img:
img, face_bboxs = self.face_detector.findFaces(self.img, draw=self.show_img) img, face_bboxs = self.face_detector.findFaces(self.img, draw=self.show_img)
if face_bboxs: if face_bboxs:
if self.show_img: if self.show_img:
self.img = img
center = face_bboxs[0]["center"] center = face_bboxs[0]["center"]
cv2.circle(self.img, center, 5, (255, 0, 255), cv2.FILLED) cv2.circle(self.img, center, 5, (255, 0, 255), cv2.FILLED)
score = face_bboxs[0]["score"][0] score = face_bboxs[0]["score"][0]
return score >= self.min_face_score return score >= self.min_face_score
else:
print('Image not usable')
return False return False
def loop(self): def loop(self):