sortion/TensorBird
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added top_k

Browse Source
This commit is contained in:
Samuel Ortion 2021-03-06 20:06:27 +01:00
parent 25190c64e2
commit 0bed68679a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/run_tf_lite_classify_bird_pict.py
View File

@ -18,7 +18,7 @@ def set_input_tensor(interpreter, image):
input_tensor = interpreter.tensor(tensor_index)()[0] input_tensor = interpreter.tensor(tensor_index)()[0]
input_tensor[:, :] = image input_tensor[:, :] = image
def classify_image(interpreter, image): def classify_image(interpreter, image, top_k=1):
"""Returns a sorted array of classification results.""" """Returns a sorted array of classification results."""
set_input_tensor(interpreter, image) set_input_tensor(interpreter, image)
interpreter.invoke() interpreter.invoke()