U ʹj@s2ddlZddlTeeZd ddZd ddZdS) N)*FcCsV|s|s|std|pi}|p"i}|s@|dk r@t|dr@|j}i}d}|dddk rb|d}n|dddk rz|d}|ddf}|dddk rt|dttfstt|ddkstt|d}|d}nt|d ddk r t|d tst||d |d f}n<|r.|d ddk r.|d }n|dddk rH|d}||d<d |d <|d drt|d |d <n|d dr|d |d <t |d <|d ddk rt|d }t|dkrtt||}nt||kst||d <n|d dr |d |d <t |d<|dddk rlt|d} t| dkrPtt| |} nt| |ksbt| |d<n|ddr|d|d<t } |ddr|d} n4|r|ddr|d} n|ddr|d} | |d<t } |ddr|d} n|ddr|d} | |d<|rRt d|D] \} } t d| t| fq0|S)NzHAt least one of model, args, or pretrained_cfg required for data config.pretrained_cfgin_chansZchans input_sizerZimg_sizeZtest_input_sizeZbicubic interpolationmeanrstdcrop_pctZ test_crop_pct crop_modez:Data processing configuration for current model + dataset:z %s: %s)AssertionErrorhasattrrget isinstancetuplelistlenintZIMAGENET_DEFAULT_MEANZIMAGENET_DEFAULT_STDZDEFAULT_CROP_PCTZDEFAULT_CROP_MODE_loggerinfoitemsstr)argsrmodel use_test_sizeverboseZ data_configrrr r r r nvr O/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/data/config.pyresolve_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 )rrrrr)r")rrrrrr r r!resolve_model_data_configgsr#)NNNFF)NNFF)logging constants getLogger__name__rr"r#r r r r!s  a