Updated README and script

This commit is contained in:
Kinar 2023-11-11 03:32:45 -08:00
parent 35f2f36733
commit 021c7edde7
2 changed files with 3 additions and 2 deletions

View File

@ -7,7 +7,7 @@ First, clone this Git repo.
Run this script to install the required dependencies and download the TFLite models:
```
cd mediapipe/tasks/python/benchmark/vision/image_classifier
cd mediapipe/mediapipe/tasks/python/benchmark/vision/image_classifier
sh setup.sh
```

View File

@ -55,7 +55,8 @@ def main():
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument(
'--model', help='Path to image classification model.', required=True)
'--model', help='Path to image classification model.', required=True,
default='classifier.tflite')
parser.add_argument(
'--iterations', help='Number of iterations for benchmarking.', type=int,
default=100)