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:
|
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
|
sh setup.sh
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
|
@ -55,7 +55,8 @@ def main():
|
||||||
parser = argparse.ArgumentParser(
|
parser = argparse.ArgumentParser(
|
||||||
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
||||||
parser.add_argument(
|
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(
|
parser.add_argument(
|
||||||
'--iterations', help='Number of iterations for benchmarking.', type=int,
|
'--iterations', help='Number of iterations for benchmarking.', type=int,
|
||||||
default=100)
|
default=100)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user