diff options
-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"], ) |