Add Raspberry Pi example

ageitgey · ageitgey · commit b7ac424986d1 · 2017-04-20T13:25:24.000-07:00
diff --git a/README.md b/README.md
@@ -71,10 +71,11 @@ See [this example](https://github.com/ageitgey/face_recognition/blob/master/exam
 
 ## Installation
 
-Python 3 / Python 2 are fully supported. Only macOS and
-Linux are tested. I have no idea if this will work on Windows. A
-[pre-configured VM](https://medium.com/@ageitgey/try-deep-learning-in-python-now-with-a-fully-pre-configured-vm-1d97d4c3e9b)
-is also available.
+Requirements:
+* Python 3+ or Python 2.7
+* macOS or Linux (Windows untested)
+* [Also can run on a Raspberry Pi 2+ (follow this specific instructions)](https://gist.github.com/ageitgey/1ac8dbe8572f3f533df6269dab35df65)
+* A [pre-configured VM image](https://medium.com/@ageitgey/try-deep-learning-in-python-now-with-a-fully-pre-configured-vm-1d97d4c3e9b) is also available.
 
 Install this module from pypi using `pip3` (or `pip2` for Python 2):
 
@@ -212,6 +213,7 @@ All the examples are available [here](https://github.com/ageitgey/face_recogniti
 * [Find and recognize unknown faces in a photograph based on photographs of known people](https://github.com/ageitgey/face_recognition/blob/master/examples/recognize_faces_in_pictures.py)
 * [Recognize faces in live video using your webcam - Simple / Slower Version (Requires OpenCV to be installed)](https://github.com/ageitgey/face_recognition/blob/master/examples/facerec_from_webcam.py)
 * [Recognize faces in live video using your webcam - Faster Version (Requires OpenCV to be installed)](https://github.com/ageitgey/face_recognition/blob/master/examples/facerec_from_webcam_faster.py)
+* [Recognize faces on a Raspberry Pi w/ camera](https://github.com/ageitgey/face_recognition/blob/master/examples/facerec_on_raspberry_pi.py)
 * [Run a web service to recognize faces via HTTP (Requires Flask to be installed)](https://github.com/ageitgey/face_recognition/blob/master/examples/web_service_example.py)
 
 ## How Face Recognition Works
diff --git a/examples/facerec_on_raspberry_pi.py b/examples/facerec_on_raspberry_pi.py
@@ -0,0 +1,48 @@
+# This is a demo of running face recognition on a Raspberry Pi.
+# This program will print out the names of anyone it recognizes to the console.
+
+# To run this, you need a Raspberry Pi 2 (or greater) with face_recognition and
+# the picamera[array] module installed.
+# You can follow this installation instructions to get your RPi set up:
+# https://gist.github.com/ageitgey/1ac8dbe8572f3f533df6269dab35df65
+
+import face_recognition
+import picamera
+import numpy as np
+
+# Get a reference to the Raspberry Pi camera.
+# If this fails, make sure you have a camera connected to the RPi and that you
+# enabled your camera in raspi-config and rebooted first.
+camera = picamera.PiCamera()
+camera.resolution = (320, 240)
+output = np.empty((240, 320, 3), dtype=np.uint8)
+
+# Load a sample picture and learn how to recognize it.
+print("Loading known face image(s)")
+obama_image = face_recognition.load_image_file("obama_small.jpg")
+obama_face_encoding = face_recognition.face_encodings(obama_image)[0]
+
+# Initialize some variables
+face_locations = []
+face_encodings = []
+
+while True:
+    print("Capturing image.")
+    # Grab a single frame of video from the RPi camera as a numpy array
+    camera.capture(output, format="bgr")
+
+    # Find all the faces and face encodings in the current frame of video
+    face_locations = face_recognition.face_locations(output)
+    print("Found {} faces in image.".format(len(face_locations)))
+    face_encodings = face_recognition.face_encodings(output, face_locations)
+
+    # Loop over each face found in the frame to see if it's someone we know.
+    for face_encoding in face_encodings:
+        # See if the face is a match for the known face(s)
+        match = face_recognition.compare_faces([obama_face_encoding], face_encoding)
+        name = "<Unknown Person>"
+
+        if match[0]:
+            name = "Barack Obama"
+
+        print("I see someone named {}!".format(name))
diff --git a/examples/obama_small.jpg b/examples/obama_small.jpg
