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/README.md | 2 +- samples/ImageClassification/run_classifier.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) (limited to 'samples') diff --git a/samples/ImageClassification/README.md b/samples/ImageClassification/README.md index 068d0c916a..e34e12a922 100644 --- a/samples/ImageClassification/README.md +++ b/samples/ImageClassification/README.md @@ -58,7 +58,7 @@ from the Arm ML-Zoo). cd BASEDIR #copy your model and label mapping cp ml-zoo/models/image_classification/mobilenet_v2_1.0_224/tflite_uint8/mobilenet_v2_1.0_224_quantized_1_default_1.tflite . - cp ml-zoo/models/image_classification/mobilenet_v2_1.0_224/tflite_uint8 labelmappings.txt . + cp ml-zoo/models/image_classification/mobilenet_v2_1.0_224/tflite_uint8/labelmappings.txt . ``` 4. Download a test image. 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