Ng8tUdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZmZddlmZmZgd ZeeZeeeefed <iZeeefed <iZ eeed e ffed <eZ!eeed<iZ"eeefed<iZ#eeefed<ee$Z%eeeefed<ee&Z'eeeeeeffed<iZ(eeeefed<d?dededeeeffdZ)dedefdZ*deeeeee fefffdZ+ded e fded e ffdZ,d@deded efd!Z-d"ed#eeeeffd$Z.d%edeee/effd&Z0d'efd(Z1 dAd'eeeefd*eeeefd+e2d,eeeefd-e2d.ee2deefd/Z3 dBd'eeeefd,edeefd0Z4d?d*edeeeffd1Z5dede2fd2Z6dCded3eeded e ffd4Z7deefd5Z8ded6eeed feeeefde2fd7Z9dede2fd8Z:dDded:e2deefd;Z;dedZ=dS)EzB Model Registry Hacked together by / Copyright 2020 Ross Wightman N) defaultdictdeque)deepcopy)replace) AnyCallableDictIterableListOptionalSetSequenceUnionTuple) PretrainedCfg DefaultCfg) split_model_name_tag get_arch_nameregister_modelgenerate_default_cfgs list_modelslist_pretrainedis_modelmodel_entrypoint list_modulesis_model_in_modulesget_pretrained_cfg_valueis_model_pretrainedget_arch_pretrained_cfgs_module_to_models_model_to_module._model_entrypoints_model_has_pretrained_model_default_cfgs_model_pretrained_cfgs_model_with_tags_module_to_deprecated_models_deprecated_models model_nameno_tagreturncR|dd^}}|r|dn|}||fS)N.rr)split)r+r,tag_listtags Q/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/models/_registry.pyrr!s8&,,S!44J! -(1++vC s?c,t|dS)Nr)rr+s r3rr's  + +A ..r4cfgscxtt}t}|D]\}}t |t r t di|}|j}t|\}}||v}|r| p| do| } | d}||} | r0| j || |n>|r"| js| j |n| j ||rd| _|| j|<|S)N*T)rrsetitems isinstancedictr has_weightsrendswithstriptags appendleftadd is_pretrainedappendr7) r7out default_setkvr?modelr2is_default_setpriority default_cfgs r3rr+sS j ! !C%%K ""1 a   #""""Am )!,, s+-+GZc1B1B1Y>GYiinn%j  )   ' ' , , , OOE " " " "  )!: )   ' ' , , , ,   # #C ( ( (  -(,K % !  Jr4fnc 4tj|j}|jd}t |r|dnd}|j}t |dr|j|n|g|_|tvr%tj d|d|jd|dd |t|<|t|<t||t |d r||jvr|j|}t!|t"sDt!|t$sJt'di|}t#t)dgd|i }t+|jD]\}}|d k} |j|}|rd||gn|} t%||r|nd } |jr|jdkr |j| z| d<t5|fi| }| r+|t6|<|jrt:||rL|t6| <|jrt:| t<|| t<|||t>|<|S)Nr/r*__all__z Overwriting z in registry with zr. This is because the name being registered conflicts with an existing name. Please check if this is not expected. stacklevel default_cfgs)rBr7r) architecturer2ztimm/ hf_hub_idr:) sysmodules __module__r0len__name__hasattrrRrFr#warningswarnr"r!rDrVr=rr>rr enumeraterBr7joinrXrr&r?r$r'r%) rOmodmodule_name_split module_namer+rNpretrained_cfgtag_idxr2 is_defaultmodel_name_tag replace_itemss r3rrKs +bm $C ++C00+./@+A+AI#B''rKJsI# :&&&&!l '''  `: ` ` ` ` ` ` `    &(z"#.Z k"&&z222sN##$6 c6F(F(F&z2 +z22 Rk400 0 0 0*99[99N$%++RRccdSSSM' WN,D,O,O-;-E-V k*$^EE}EEN :5C&z2!-:)--j999 @9G&~6!->)--n=== ,33NCCCC ,33J????*5J' Ir4deprecated_name current_fn current_tagcdfd }|S)NFcJddrdjgnj}tjdd|dd|dd}d ||pd |S) NzModel z& has been removed with no replacement.r/zMapping deprecated model name z to current rSrTrf) pretrainedrfr:)rbr]r_r`pop)rpkwargs current_namerfrlrmrks r3_fnz#_deprecated_model_shim.._fns%%(h(h(h(h%%%GRksxx!4k BCCCXbXk  cccT`cccpqrrrr$4d;;zhZ@]R]hhaghhhr4)Fr:)rkrlrmrts``` r3_deprecated_model_shimrus<iiiiiiii Jr4redeprecation_mapc>tj|}|d}t|r|dnd}|D]\}}t |dr|j|d}d}|r"t|\}}t||}t|||} t||| | t|<|t|<t|||t |<|t"||<dS)Nr/rQr*rR)rYrZr0r\r<r^rRrFrgetattrrusetattrr#r"r!rDr)r() rervrcrd deprecatedcurrentrlrmrsdeprecated_entrypoint_fns r3register_model_deprecationsr}s< +k "C#))#..+./@+A+AI#B''rK.4466HH G 3 " " + K  z * * *   4(z _natural_key..s/ V V VQaiikk (CFFFq V V Vr4z(\d+))rer0lower)r~s r3 _natural_keyrs* V V"(8W]]__2U2U V V VVr4filtercbt|\}}|sd|dg|gS|gS)z< expand a 'base_filter' to 'base_filter.*' if no tag portionr/r9)rrb)r filter_base filter_tags r3_expand_filterrs?26::K +s+,,f55xr4Fmodulerpexclude_filtersname_matches_cfg include_tagscn|r"t|ttfr|n|g}ng}||}|s'tt}nmt|t rt|}nJt|tsJt}|D]"}| t|#|tz }|rMt} |D]"}| t|#| }d|D}d|D}|rMt} |D];} tj || } t| r| | } ..SSS!PQARARSS22SSSSr4c6g|]}t|D]}|Sr:rrs r3rzlist_models..rr4)key)r=tuplelistr;r#keysstrr!rupdater)r'fnmatchrr\union differencer$ intersectionr&sortedr)rrrprrrinclude_filters all_modelsmmodels_with_tagsmodelsrinclude_modelsxfexclude_modelss r3rrsf4$.vt}$E$ES&&F8!  8"#5#:#:#<#<== fc " " 8#4V#"" 6n55 6 ;/E4=99 0./O! ; ;B$^FB77N>"" ;**>::<&33F;;B+,,99&AA &l + + ++r4c(t|d|dS)NT)rrprr)r)rrs r3rr s' '    r4cL|r t|nt}t|SN)r(r)r)rall_deprecateds r3get_deprecated_modelsrs&=C[1&99I[N N # ##r4c2t|}|tvS)z" Check if a model name exists )rr#)r+ arch_names r3rrsj))I * **r4 module_filterct|}|r3|t|ivrtd|d|dt|S)z6Fetch a model entrypoint for specified model name zModel (z not found in module r/)rr!get RuntimeErrorr#)r+rrs r3rr$sbj))IX*;*?*? r*R*RRRVZVVmVVVWWW i ((r4cRt}t|S)zI Return list of module names that contain models / model entrypoints )r!rr)rZs r3rr-s! $$&&G '??r4 module_namesct|t|tttfsJt fd|DS)zCheck if a model exists within a subset of modules Args: model_name - name of model to check module_names - names of modules to search in c36K|]}t|vVdSr)r!)rnrs r3 z&is_model_in_modules..?s.GGQy-a00GGGGGGr4)rr=rrr;any)r+rrs @r3rr4sRj))I lUD#$6 7 7777 GGGG,GGG G GGr4c|tvSr)r$r6s r3rrBs . ..r4Tallow_unregisteredc|tvrtt|St|\}}|tvrt d|d|d|rdSt d|d)NzInvalid pretrained tag (z) for r/zModel architecture (z#) has no pretrained cfg registered.)r&rrr%r)r+rrr2s r3get_pretrained_cfgrFs+++.z:;;;)*55NIs'''McMMMMMNNNt \i\\\ ] ]]r4cfg_keycFt|d}t||dS)zO Get a specific model default_cfg value by key. None if key doesn't exist. F)rN)rrx)r+rcfgs r3rrSs( ZE B B BC 3 & &&r4c\t|\}}t|}d|D}|S)z7 Get all pretrained cfgs for a given architecture. c*i|]}|t|Sr:)r&)rrs r3 z,get_arch_pretrained_cfgs.._s > > >QA%a( > > >r4)rr')r+r_ model_namesr7s r3r r Zs7( 33LIq"9-K > >+ > > >D Kr4)r*)Nr*)r*r*Fr*FN)r*r*r)T)>__doc__rrrYr_ collectionsrrcopyr dataclassesrtypingrrr r r r r rrr _pretrainedrrrRr;r!r__annotations__r"r#r$r%r&rr'r>r(r)rrrrrur}rrrboolrrrrrrrrrrr r:r4r3rsJ ********]]]]]]]]]]]]]]]]]]]]]]]]22222222    *5S)9)94SX &999#%$sCx.%%%46DhsCx001666"%#%%s3x'''02T#},-22235S-/0555)4T):):$sDI~&:::DOKPTDUDUd3S(3--?(@#@AUUU/1Dhsm+,111S#uS#X /c/c////S%S#X 0M*N%N O@=xS)=hsCx.@====@CX[^HSH4XVY]HZC[HHHH,W#W$uS#X"7WWWW 3)+(* 13!&'+ P,P,c49n%P,c49n%P,P,sDI~. P,  P, tn P, #YP,P,P,P,h)+!  c49n%   #Y    $$#$tCH~$$$$ ++++++)))Xc])hWZ\_W_N`))))d3i H H',U38_d3iS-Q'R H  H H H H/C/D//// ^ ^3 ^D ^HUbLc ^ ^ ^ ^''s'x}''''c=6H1Ir4