From 5ed29208787f265b096f1ce30e3a91a4411e8452 Mon Sep 17 00:00:00 2001 From: Davide Tonelli Date: Tue, 9 Aug 2022 12:52:01 +0200 Subject: [PATCH] resulting labels are filtered and sorted --- vision-env/src/vision-test.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/vision-env/src/vision-test.py b/vision-env/src/vision-test.py index e2237f7..dae899b 100644 --- a/vision-env/src/vision-test.py +++ b/vision-env/src/vision-test.py @@ -6,6 +6,9 @@ import cv2 # Imports the Google Cloud client library from google.cloud import vision +# Minimum score required for labels detection +MIN_SCORE_REQUIRED = 0.80 + def shape_selection(event, x, y, flags, param): # grab references to the global variables global ref_point, crop @@ -85,6 +88,12 @@ image = vision.Image(content=content) response = client.label_detection(image=image) labels = response.label_annotations +# Labels sorted by attribute score +sorted_labels = sorted(labels, key=lambda x:x.score, reverse=True) + +# Labels filtered by attribute score +filtered_labels = filter(lambda x:x.score > MIN_SCORE_REQUIRED, sorted_labels) + print('Labels:') -for label in labels: - print(label.description) +for label in filtered_labels: + print(label.description + ' ---> ' + str(label.score))