tol/vision-engine-layer-1
Archived
1
0
Fork 0
Code Issues Activity

Merge pull request #10 from DavideTonel/main

Browse Source 
Resulting labels are filtered and sorted
This commit is contained in:
Davide Tonelli
2022-08-09 12:53:58 +02:00
committed by GitHub
parent ab23dcc46c 5ed2920878
commit 77a25c8d0a
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vision-env/src/vision-test.py
+11 -2
View File
@@ -6,6 +6,9 @@ import cv2
# Imports the Google Cloud client library # Imports the Google Cloud client library
from google.cloud import vision from google.cloud import vision
# Minimum score required for labels detection
MIN_SCORE_REQUIRED = 0.80
def shape_selection(event, x, y, flags, param): def shape_selection(event, x, y, flags, param):
# grab references to the global variables # grab references to the global variables
global ref_point, crop global ref_point, crop
@@ -85,6 +88,12 @@ image = vision.Image(content=content)
response = client.label_detection(image=image) response = client.label_detection(image=image)
labels = response.label_annotations 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:') print('Labels:')
for label in labels: for label in filtered_labels:
print(label.description) print(label.description + ' ---> ' + str(label.score))