in TSA-demo/TSA_rekognition/detect_labels_image.py [0:0]
def detect_labels_local_file(photo):
fullreturnarray = []
keyreturnarray = []
tsaimage = []
with open(photo, 'rb') as image:
response = client.detect_labels(Image={'Bytes': image.read()})
print('Detected labels in ' + photo)
for label in response['Labels']:
print (label['Name'] + ' : ' + str(label['Confidence']))
tsaimage = get_tsaimage(label['Name'])
if label['Confidence'] < 40:
print('recognition confidence below 40')
break
if tsaimage:
print("Get tsa image succeeded for " + label['Name'])
partialCat = tsaimage[0]["id"]["S"]
print(partialCat)
exists = partialCat in keyreturnarray
if exists == False :
fullreturnarray.append(tsaimage[0])
keyreturnarray.append(partialCat)
return fullreturnarray