o èe)ã@s6ddlZddlZdadaGdd„deƒZddd„ZdS)éNc@s(eZdZdZdd„Zdd„Zdd„ZdS) Ú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. cCs4t|tƒsJ‚ddgddddddœ|_| |¡dS) Né éZAllÚtotalZDefaultz /tmp/profileT)Ú batch_rangeÚstateÚ sorted_keyÚ tracer_optionÚ profile_pathÚexit_on_finished)Ú isinstanceÚstrÚ_optionsÚ_parse_from_string)ÚselfÚ options_str©rúND:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/ppocr/utils/profiler.pyÚ__init__.súzProfilerOptions.__init__cCs¼| dd¡ d¡D]R}| d¡\}}|dkrD| dd¡ dd¡ d¡}ttt|ƒƒ}t|ƒd krC|d d krC|d |d krC||j|<q |d krR| ¡d v|j|<q |dvr[||j|<q dS)Nú Úú;ú=rú[ú]ú,érér )ÚyesÚtrueÚtÚ1)rrr r )ÚreplaceÚsplitÚlistÚmapÚintÚlenrÚlower)rrÚkvÚkeyÚvalueZ value_listrrrr;s$ÿÿ € €óz"ProfilerOptions._parse_from_stringcCs(|j |d¡durtd|ƒ‚|j|S)Nz1ProfilerOptions does not have an option named %s.)rÚgetÚ ValueError)rÚnamerrrÚ __getitem__Ks ÿ zProfilerOptions.__getitem__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr/rrrrrs  rcCsŒ|durdStdurt|ƒattddkr#tjj tdtd¡nttddkr@tjj tdtd¡td r@t  d¡td7adS) al 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. Nrrrr rrr r ) Ú_profiler_optionsrÚ_profiler_step_idÚpaddleÚutilsZprofilerZstart_profilerZ stop_profilerÚsysÚexit)rrrrÚadd_profiler_stepRs  ÿÿ  r:)N)r8r6r5r4Úobjectrr:rrrrÚs 8