Skip to content

Commit

Permalink
Display the score/confidence value (#1429)
Browse files Browse the repository at this point in the history
* Display the score/confidence value

A small code addition to display the score/confidence value of a detected face above the face detection box on the output image. This is very useful to know the confidence!

* Changes applied to meet coding style requirements 

I have edited the already submitted code to meet the coding style requirements!

* Edits because white spaces
  • Loading branch information
peshmerge authored and engelke committed Nov 19, 2018
1 parent 614e2f2 commit 0b0d619
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions vision/cloud-client/face_detection/faces.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
# [START vision_face_detection_tutorial_imports]
from google.cloud import vision
from google.cloud.vision import types
from PIL import Image, ImageDraw
from PIL import Image, ImageDraw, ImageFont
# [END vision_face_detection_tutorial_imports]


Expand Down Expand Up @@ -59,12 +59,18 @@ def highlight_faces(image, faces, output_filename):
"""
im = Image.open(image)
draw = ImageDraw.Draw(im)

# Sepecify the font-family and the font-size
font = ImageFont.truetype("arial.ttf", 25)
for face in faces:
box = [(vertex.x, vertex.y)
for vertex in face.bounding_poly.vertices]
draw.line(box + [box[0]], width=5, fill='#00ff00')

# Place the confidence value/score of the detected faces above the
# detection box in the output image
draw.text(((face.bounding_poly.vertices)[0].x,
(face.bounding_poly.vertices)[0].y - 30),
str(format(face.detection_confidence, '.3f')) + '%',
font=font, fill='#FF0000')
im.save(output_filename)
# [END vision_face_detection_tutorial_process_response]

Expand Down

0 comments on commit 0b0d619

Please sign in to comment.