From 0bed68679af0305fb97b4ff152c1a516cfa1e3ef Mon Sep 17 00:00:00 2001 From: UncleSamulus Date: Sat, 6 Mar 2021 20:06:27 +0100 Subject: [PATCH] Added top_k --- src/run_tf_lite_classify_bird_pict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/run_tf_lite_classify_bird_pict.py b/src/run_tf_lite_classify_bird_pict.py index caa2a54..ffc83df 100644 --- a/src/run_tf_lite_classify_bird_pict.py +++ b/src/run_tf_lite_classify_bird_pict.py @@ -18,7 +18,7 @@ def set_input_tensor(interpreter, image): input_tensor = interpreter.tensor(tensor_index)()[0] input_tensor[:, :] = image -def classify_image(interpreter, image): +def classify_image(interpreter, image, top_k=1): """Returns a sorted array of classification results.""" set_input_tensor(interpreter, image) interpreter.invoke()