a  d/@sddlZddlZddlZddlZddlZddlmZddlm Z ddZ ddZ d d Z e d d Z dddZddZdddZdS)N)path) master_onlycCs<t|tj|t|tj|tj|dS)zSet random seeds.N)randomseednptorchZ manual_seedZcudaZmanual_seed_all)rr [/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/utils/misc.pyset_random_seed s     r cCstdtS)Nz %Y%m%d_%H%M%S)timestrftime localtimer r r r get_time_strsrcCsHt|r6|dt}td|ddt||tj|dddS)zwmkdirs. If path exists, rename it with timestamp and create a new one. Args: path (str): Folder path. Z _archived_z"Path already exists. Rename it to T)flushexist_okN)ospexistsrprintosrenamemakedirs)rnew_namer r r mkdir_and_renames   rcCsz|d}|dr$t|dnt|d|D]:\}}d|vs:d|vs:d|vs:d|vrfq:q:tj|d d q:d S) zMake dirs for experiments.rZis_trainZexperiments_rootZ results_rootZ strict_loadZpretrain_networkresume param_keyTrN)copyrpopitemsrr)optZpath_optkeyrr r r make_exp_dirs%s  r"Fcs@|durt|ttfstd|fdd|||dS)aScan a directory to find the interested files. Args: dir_path (str): Path of the directory. suffix (str | tuple(str), optional): File suffix that we are interested in. Default: None. recursive (bool, optional): If set to True, recursively scan the directory. Default: False. full_path (bool, optional): If set to True, include the dir_path. Default: False. Returns: A generator for all the interested files with relative paths. Nz-"suffix" must be a string or tuple of stringsc3st|D]r}|jds^|r^r.|j}nt|j}|durL|Vq|||r||Vq |r |j||dEdHq q q dS)N.suffix recursive) rscandirname startswithis_filerrrelpathendswith)dir_pathr%r&entryZ return_path_scandir full_pathrootr r r0Is zscandir.._scandirr$) isinstancestrtuple TypeError)r-r%r&r1r r/r r'4s r'c Cs.|ddr*dd|D}d}|D] }|dd|dur(d}q(|rVtd |D]}d|}|d d }|dd dus||dd vrZt|dd d|d|d|d|<td|d|d|qZdd|dD}|D]0}|d|dkrd|d|<td|dqdS)zCheck resume states and pretrain_network paths. Args: opt (dict): Options. resume_iter (int): Resume iteration. rZ resume_statecSsg|]}|dr|qS)network_r).0r!r r r gz check_resume..FZ pretrain_NTz6pretrain_network path will be ignored during resuming.r7Zignore_resume_networksmodelsZnet__z.pthzSet z to cSsg|]}|dr|qS)rr8r9r r r r;xr<Z params_emaparamsz to params)keysgetrreplacerjoin) r Z resume_iterZnetworksZ flag_pretrainnetworkr(basenameZ param_keysrr r r check_resume^s*   * rGBcCsFdD]0}t|dkr,|dd||S|d}q|dd|S)zGet human readable file size. Args: size (int): File size. suffix (str): Suffix. Default: 'B'. Return: str: Formatted file size. )r=KMGTPEZg@z3.1f z Y)abs)sizer%unitr r r sizeof_fmts   rT)NFF)rH)numpyrrrr rrrZ dist_utilrr rrr"r'rGrTr r r r s      *!