From 252cca72e77c13e5f284404a6b4c2b52e08369ca Mon Sep 17 00:00:00 2001 From: Kinar Date: Mon, 13 Nov 2023 21:44:27 -0800 Subject: [PATCH] Allowed a default value for the model argument --- .../vision/image_classifier/image_classifier_benchmark.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 d505a8fe8..65a629bab 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 @@ -46,7 +46,7 @@ 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=False, default='classifier.tflite') parser.add_argument( '--iterations', help='Number of iterations for benchmarking.', type=int,