From f1c7f0020874ada342cacf301b1d137e67751deb Mon Sep 17 00:00:00 2001 From: Henri Woodcock Date: Wed, 26 May 2021 17:37:18 +0100 Subject: CSAF-235 Small fixes to Arm NN Delegate Image Classificaton Example Signed-off-by: Henri Woodcock henri.woodcock@arm.com Change-Id: If3aed2e50c82418e016c78479896f987961b0e9f --- samples/ImageClassification/run_classifier.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'samples/ImageClassification/run_classifier.py') diff --git a/samples/ImageClassification/run_classifier.py b/samples/ImageClassification/run_classifier.py index 1b4b9ed61e..b3f79ea63c 100644 --- a/samples/ImageClassification/run_classifier.py +++ b/samples/ImageClassification/run_classifier.py @@ -147,7 +147,7 @@ def create_mapping(label_mapping_p): with open(label_mapping_p) as label_mapping_raw: for line in label_mapping_raw: label_mapping[idx] = line - idx = 1 + idx += 1 return label_mapping @@ -228,7 +228,7 @@ if __name__ == "__main__": "--preferred_backends", help="list of backends in order of preference", type=str, - nargs=" ", + nargs="+", required=False, default=["CpuAcc", "CpuRef"], ) -- cgit v1.2.1