diff options
author | Henri Woodcock <henri.woodcock@arm.com> | 2021-05-26 17:37:18 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-05-27 10:15:56 +0000 |
commit | f1c7f0020874ada342cacf301b1d137e67751deb (patch) | |
tree | 01e51d50d0c1d42b5c1e5bccdaed5616039428b9 /samples/ImageClassification | |
parent | 94dd4dbbb2b59f47e38cf2f1280c4b25be45ea64 (diff) | |
download | armnn-f1c7f0020874ada342cacf301b1d137e67751deb.tar.gz |
CSAF-235 Small fixes to Arm NN Delegate Image Classificaton Example
Signed-off-by: Henri Woodcock henri.woodcock@arm.com
Change-Id: If3aed2e50c82418e016c78479896f987961b0e9f
Diffstat (limited to 'samples/ImageClassification')
-rw-r--r-- | samples/ImageClassification/README.md | 2 | ||||
-rw-r--r-- | samples/ImageClassification/run_classifier.py | 4 |
2 files changed, 3 insertions, 3 deletions
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"], ) |