diff --git a/mediapipe/tasks/python/benchmark/vision/image_classifier/README.md b/mediapipe/tasks/python/benchmark/vision/image_classifier/README.md index 0c3e23432..d7db3559d 100644 --- a/mediapipe/tasks/python/benchmark/vision/image_classifier/README.md +++ b/mediapipe/tasks/python/benchmark/vision/image_classifier/README.md @@ -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 ``` diff --git a/mediapipe/tasks/python/benchmark/vision/image_classifier/image_classifier_benchmark.py b/mediapipe/tasks/python/benchmark/vision/image_classifier/image_classifier_benchmark.py index c6a80da11..3fc7c8a2e 100644 --- a/mediapipe/tasks/python/benchmark/vision/image_classifier/image_classifier_benchmark.py +++ b/mediapipe/tasks/python/benchmark/vision/image_classifier/image_classifier_benchmark.py @@ -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)