Updated README and script
This commit is contained in:
parent
35f2f36733
commit
021c7edde7
|
@ -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
|
||||
```
|
||||
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue
Block a user