uki :dZddlZddlmZddlmZddlmZdZ y)zMediaPipe vision benchmarker.N)benchmark_utils)base_vision_benchmark_api) base_optionsc tjtj}|jdddd|jddd |jd d td |jddt d|j }tjtj|}tj|j|}d|jiid}|j|vrtd|j||j}i}t j"j$j&t j"j$j(fD]g} |||j*| } |jdk(rtj,| fi||| <@|jdk(sPtj.| || <i|j1D](\} } t3d| d||jd| dd*y )aExecutes a benchmarking process using a specified function ann model. Args: benchmark_function: A callable function to be executed for benchmarking. This function should contain the logic of the task to be benchmarked and should be capable of utilizing a model specified by its name. default_model_name: The name or path of the default model to be used in the benchmarking process. This is useful when the benchmarking function requires a model and no other model is explicitly specified. )formatter_classz--modez+Benchmarking mode (e.g., "nth_percentile").Fnth_percentile)helprequireddefaultz--modelzPath to the model.N)r r z --iterationsz&Number of iterations for benchmarking.d)r typer z --percentilez'Percentile for benchmarking statistics.gW@ percentile)raveragezUnsupported benchmarking mode: rzInference time  z: z.6fz milliseconds)argparseArgumentParserArgumentDefaultsHelpFormatter add_argumentintfloat parse_argsbuget_test_data_pathrVISION_TEST_DATA_DIRget_model_pathmodelrmode ValueErrorr BaseOptionsDelegateCPUGPU iterationsrritemsprint) benchmark_functiondefault_model_nameparserargsdefault_model_path model_pathmode_args_mapping mode_argsresults delegate_typeinference_timesresults l/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/benchmark/vision/benchmark.py benchmarkerr3s  " "<< &  8   i&:DI 3     4      $,,446H  -?@*&t7  YY'' 6tyykB CC *) '''++''++ ;m)DOO]O  yy$$!00NINgm i !zz/:gm ; '}}mV -*;DII*F)Gr #,m %) __doc__r mediapipe.tasks.python.benchmarkrr,mediapipe.tasks.python.benchmark.vision.corermediapipe.tasks.python.corerr3r4r2r:s$BR4Lr4