From 78bfdda45c191ed0b79c2ecb3a3ce3301f7cec59 Mon Sep 17 00:00:00 2001 From: Davide Tonelli Date: Sat, 23 Jul 2022 14:05:26 +0200 Subject: [PATCH] Script takes image path as input --- vision-env/src/vision-test.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/vision-env/src/vision-test.py b/vision-env/src/vision-test.py index b9dbd9f..794f8da 100644 --- a/vision-env/src/vision-test.py +++ b/vision-env/src/vision-test.py @@ -1,19 +1,23 @@ import io import os +import sys # Imports the Google Cloud client library from google.cloud import vision -# Instantiates a client -client = vision.ImageAnnotatorClient() +# User input from the command line +image_path = sys.argv[1] # The name of the image file to annotate -file_name = os.path.abspath('images/mmr.jpg') +file_name = os.path.abspath(image_path) # Loads the image into memory with io.open(file_name, 'rb') as image_file: content = image_file.read() +# Instantiates a client +client = vision.ImageAnnotatorClient() + image = vision.Image(content=content) # Performs label detection on the image file