o ÓÙ¾iã@sDddlZddlTe e¡Z     d dd„Z    d dd„ZdS) éNé)Ú*FcCsB|s |s |s Jdƒ‚|p i}|pi}|s |dur t|dƒr |j}i}d}| dd¡dur1|d}n | dd¡dur=|d}|ddf}| dd¡durjt|dttfƒsUJ‚t|dƒdks_J‚t|dƒ}|d}n6| d d¡dur…t|d tƒs{J‚||d |d f}n|r”| d d¡dur”|d }n | dd¡dur |d}||d<d |d <| d d¡rµ|d |d <n | d d¡rÁ|d |d <t|d <| d d¡durït|d ƒ}t|ƒdkrâtt|ƒ|ƒ}nt|ƒ|ksêJ‚||d <n | d d¡rû|d |d <t |d<| dd¡dur,t|dƒ} t| ƒdkrtt| ƒ|ƒ} n t| ƒ|ks'J‚| |d<n | dd¡r9|d|d<t } | dd¡rG|d} n|rV| dd¡rV|d} n | dd¡ra|d} | |d<t } | dd¡rs|d} n | dd¡r~|d} | |d<|rŸt   d¡| ¡D]\} } t   d| t| ƒf¡qŽ|S)NzHAt least one of model, args, or pretrained_cfg required for data config.Úpretrained_cfgéÚin_chansÚchanséàÚ input_sizerÚimg_sizeÚtest_input_sizeÚbicubicÚ interpolationÚmeanrÚstdÚcrop_pctÚ test_crop_pctÚ crop_modez:Data processing configuration for current model + dataset:z %s: %s)ÚhasattrrÚgetÚ isinstanceÚtupleÚlistÚlenÚintÚIMAGENET_DEFAULT_MEANÚIMAGENET_DEFAULT_STDÚDEFAULT_CROP_PCTÚDEFAULT_CROP_MODEÚ_loggerÚinfoÚitemsÚstr)ÚargsrÚmodelÚ use_test_sizeÚverboseÚ data_configrr rrrrÚnÚv©r)úD/home/ubuntu/.local/lib/python3.10/site-packages/timm/data/config.pyÚresolve_data_configs†                    r+cCst|||||dS)ab Resolve Model Data Config This is equivalent to resolve_data_config() but with arguments re-ordered to put model first. Args: model (nn.Module): the model instance args (dict): command line arguments / configuration in dict form (overrides pretrained_cfg) pretrained_cfg (dict): pretrained model config (overrides pretrained_cfg attached to model) use_test_size (bool): use the test time input resolution (if one exists) instead of default train resolution verbose (bool): enable extra logging of resolved values Returns: dictionary of config )r"rr#r$r%)r+)r#r"rr$r%r)r)r*Úresolve_model_data_configgsûr,)NNNFF)NNFF)ÚloggingÚ constantsÚ getLoggerÚ__name__rr+r,r)r)r)r*Ús  ûaû