cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated README and script

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mediapipe/tasks/python/benchmark/vision/image_classifier/README.md
View File

@ -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
``` ```

3
mediapipe/tasks/python/benchmark/vision/image_classifier/image_classifier_benchmark.py
View File

@ -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)