a  þdÓ ã@s.ddlZdd„Zd dd„Zdd„Zd d „ZdS) éNcCsbddlm}||ƒ}| ¡| ¡d}td|ƒdd„| |¡Dƒ}dd„| |¡Dƒ}||fS)z Get raw data (steps and values) from tensorboard events. Args: log_path (str): Path to the tensorboard log. tag (str): tag to be read. r)ÚEventAccumulatorZscalarsz tag list: cSsg|]}t|jƒ‘qS©)ÚintÚstep©Ú.0Úsrrú`/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/utils/plot_util.pyÚ óz.read_data_from_tensorboard..cSsg|] }|j‘qSr)Úvaluerrrr r r )Z6tensorboard.backend.event_processing.event_accumulatorrZReloadZTagsÚprintZScalars)Zlog_pathÚtagrZ event_accZ scalar_listÚstepsÚvaluesrrr Úread_data_from_tensorboards   rFc Csªt|ƒ}| ¡}Wdƒn1s&0Ydd„|Dƒ}g}g}t |¡}|D]:}| |¡}|rT| t| d¡ƒ¡| t| d¡ƒ¡qT|r¢dd„|Dƒ}||fS)zíRead data from txt with 2 returned values (usually [step, value]). Args: path (str): path to the txt file. pattern (str): re (regular expression) pattern. step_one (bool): add 1 to steps. Default: False. NcSsg|] }| ¡‘qSr©Ústrip©rÚlinerrr r !r z)read_data_from_txt_2v..éécSsg|] }|d‘qS)rr)rÚvrrr r ,r ) ÚopenÚ readlinesÚreÚcompileÚmatchÚappendrÚgroupÚfloat) ÚpathÚpatternZstep_oneÚfÚlinesrrrrrrr Úread_data_from_txt_2vs &  r%cCs|t|ƒ}| ¡}Wdƒn1s&0Ydd„|Dƒ}g}t |¡}|D]&}| |¡}|rP| t| d¡ƒ¡qP|S)zœRead data from txt with 1 returned values. Args: path (str): path to the txt file. pattern (str): re (regular expression) pattern. NcSsg|] }| ¡‘qSrrrrrr r 9r z)read_data_from_txt_1v..r)rrrrrrr r)r!r"r#r$Údatarrrrr Úread_data_from_txt_1v0s &  r'cCs<g}|d}|D]&}||d||}| |¡|}q|S)ag Smooth data using 1st-order IIR low-pass filter (what tensorflow does). Ref: https://github.com/tensorflow/tensorboard/blob/f801ebf1f9fbfe2baee1ddd65714d0bccc640fb1/ tensorboard/plugins/scalar/vz_line_chart/vz-line-chart.ts#L704 Args: values (list): A list of values to be smoothed. smooth_weight (float): Smooth weight. rr)r)rZ smooth_weightZ values_smZ last_sm_valuer Zvalue_smrrr Ú smooth_dataDs  r()F)rrr%r'r(rrrr Ús