Skip to content

Instantly share code, notes, and snippets.

@thepycoder
Last active December 5, 2018 15:09
Show Gist options
  • Save thepycoder/759589565d6de72fed015e62e8844d99 to your computer and use it in GitHub Desktop.
Save thepycoder/759589565d6de72fed015e62e8844d99 to your computer and use it in GitHub Desktop.
client = vision.ImageAnnotatorClient()
def mask_faces(path):
"""Detects faces in an image."""
with io.open(path, 'rb') as image_file:
content = image_file.read()
image = vision.types.Image(content=content)
response = client.face_detection(image=image)
faces = response.face_annotations
source_img = Image.open(path)
draw = ImageDraw.Draw(source_img)
for face in faces:
v = face.bounding_poly.vertices
box = [v[0].x, v[0].y, v[2].x, v[2].y]
draw.rectangle(box, fill='black')
base_path = os.path.basename(path)
anon_path = os.path.splitext(base_path)[0]
source_img.save(os.path.join("anon_images", "%s.jpg") % anon_path, "JPEG")
mask_faces(os.path.join("images", "car_and_person.jpg"))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment