o 'j@s>ddlZddlmZdadadaGdddeZdddZdS)Nc@s(eZdZdZddZddZddZdS) ProfilerOptionsa Use a string to initialize a ProfilerOptions. The string should be in the format: "key1=value1;key2=value;key3=value3". For example: "profile_path=model.profile" "batch_range=[50, 60]; profile_path=model.profile" "batch_range=[50, 60]; tracer_option=OpDetail; profile_path=model.profile" ProfilerOptions supports following key-value pair: batch_range - a integer list, e.g. [100, 110]. state - a string, the optional values are 'CPU', 'GPU' or 'All'. sorted_key - a string, the optional values are 'calls', 'total', 'max', 'min' or 'ave. tracer_option - a string, the optional values are 'Default', 'OpDetail', 'AllOpDetail'. profile_path - a string, the path to save the serialized profile data, which can be used to generate a timeline. exit_on_finished - a boolean. cCs6t|tsJddgddddddd|_||dS) N ZAlltotalZDefaultz /tmp/profileT) batch_rangestate sorted_key tracer_option profile_pathexit_on_finished timer_only) isinstancestr_options_parse_from_string)self options_strr_/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/ppocr/utils/profiler.py__init__/s zProfilerOptions.__init__cCs|dddD]\}|d\}}|dkrD|ddddd}ttt|}t|d krC|d d krC|d |d krC||j|<q |d krR|d v|j|<q |dvr\||j|<q |dkre||j|<q dS)N ;=r[],rr )yestruet1)rrr r r )replacesplitlistmapintlenrlower)rrkvkeyvalueZ value_listrrrr=s(   z"ProfilerOptions._parse_from_stringcCs(|j|ddurtd||j|S)Nz1ProfilerOptions does not have an option named %s.)rget ValueError)rnamerrr __getitem__Os  zProfilerOptions.__getitem__N)__name__ __module__ __qualname____doc__rrr0rrrrrs  rcCs|durdStdurt|atdur6ttdtdk}tjtddtddftd|datntt tddkrYt tj dd d d datd rYt dt d7a dS) ag Enable the operator-level timing using PaddlePaddle's profiler. The profiler uses a independent variable to count the profiler steps. One call of this function is treated as a profiler step. Args: profiler_options - a string to initialize the ProfilerOptions. Default is None, and the profiler is disabled. Nr Trrrz./profiler_log)Z schedulerZon_trace_readyr Fms)Z op_detailZ thread_sepZ time_unitr )_profiler_optionsr_profrprofilerZProfilerZexport_chrome_tracingstartstep_profiler_step_idstopsummarysysexit)rZ _timer_onlyrrradd_profiler_stepVs0    r@)N) r>Zpaddle.profilerr8r;r6r7objectrr@rrrrs <