U ʹj% @sdZddlmZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZGd d d ejZd2d d Zd3ddZe eddeddeddeddddeddddddeddddddedddeddddZe d4ed d!d"Ze d5ed d#d$Ze d6ed d%d&Ze d7ed d'd(Ze d8ed d)d*Ze d9ed d+d,Ze d:ed d-d.Ze d;ed d/d0Zd1S)tt||d|}| pX| } |rt|dksj| rt|}d}nd}||_tj||ddd|_|||_ |dd|_ |dkr| rtj d|dd nd|_ |jdkrt ||d|| | ||||d |_t|_t|_t|_nLtj||d|| | |dd |_|||_|dk rJ|nt|_|dd|_|dkr| stj d|dd nd|_tj||d ddd|_||d |_|dd|_||_dS) NrgP@rF) kernel_sizebiasT)Zinplace)padding)rstriderdilationgroupsradix norm_layerZ drop_layer)rrrrrrr )superr __init__AssertionErrorintrrZConv2dconv1bn1act1Z AvgPool2d avd_firstrconv2ZIdentitybn2 drop_blockact2avd_lastconv3bn3act3 downsample)selfZinplanesZplanesrr(r cardinality base_widthavdrZis_firstZ reduce_firstrZfirst_dilationZ act_layerrZ attn_layerZaa_layerr"Z drop_pathZ group_widthZ avd_stride __class__R/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/models/resnest.pyrsd      "      & zResNestBottleneck.__init__cCs&t|jdddk r"tj|jjdS)Nweight)getattrr&rinitZzeros_r1)r)r/r/r0zero_init_lastWsz ResNestBottleneck.zero_init_lastcCs|}||}||}||}|jdk r6||}||}||}||}||}|jdk rr||}| |}| |}|j dk r| |}||7}| |}|SN) rrrrr r!r"r#r$r%r&r(r')r)xZshortcutoutr/r/r0forward[s&                zResNestBottleneck.forward) __name__ __module__ __qualname____doc__Z expansionrZReLUZ BatchNorm2drr4r8 __classcell__r/r/r-r0r s,?r FcKstt||f|Sr5)rr )variant pretrainedkwargsr/r/r0_create_resnestxsrAc Ks|dddddttddd |S) Ni)rrC)rDg?Zbilinearzconv1.0fc) urlZ num_classes input_size pool_sizecrop_pct interpolationZmeanZstdZ first_conv classifierr)rFr@r/r/r0_cfgsrLztimm/) hf_hub_id)rrN)rO)rMrGrH)r@rP) rQgJ +?Zbicubic)rMrGrHrIrJ)rrR) rSgV-?)rMrJ)zresnest14d.gluon_in1kzresnest26d.gluon_in1kzresnest50d.in1kzresnest101e.in1kzresnest200e.in1kzresnest269e.in1kzresnest50d_4s2x40d.in1kzresnest50d_1s4x24d.in1k)returnc KsDttddddgdddddtddddd }td d |it|f|S) z5 ResNeSt-14d model. Weights ported from GluonCV. rdeep TrFrr,rblockZlayersZ stem_typeZ stem_widthZavg_downr+r*Z block_args resnest14dr?)r[dictr rAr?r@Z model_kwargsr/r/r0r[s  r[c KsDttddddgdddddtddddd }td d |it|f|S) z5 ResNeSt-26d model. Weights ported from GluonCV. rWrUrVTrrFrXrY resnest26dr?)r_r\r^r/r/r0r_s  r_c KsDttddddgdddddtd dd d d }tdd|it|f|S)z ResNeSt-50d model. Matches paper ResNeSt-50 model, https://arxiv.org/abs/2004.08955 Since this codebase supports all possible variations, 'd' for deep stem, stem_width 32, avg in downsample. rr rUrVTrrrWFrXrY resnest50dr?)rar\r^r/r/r0ras  rac KsDttddddgdddddtddd d d }tdd |it|f|S)z ResNeSt-101e model. Matches paper ResNeSt-101 model, https://arxiv.org/abs/2004.08955 Since this codebase supports all possible variations, 'e' for deep stem, stem_width 64, avg in downsample. rr rUrTrrWFrXrY resnest101er?)rcr\r^r/r/r0rcs  rcc KsDttddddgdddddtddd d d }tdd |it|f|S)z ResNeSt-200e model. Matches paper ResNeSt-200 model, https://arxiv.org/abs/2004.08955 Since this codebase supports all possible variations, 'e' for deep stem, stem_width 64, avg in downsample. r$rUrTrrWFrXrY resnest200er?)rfr\r^r/r/r0rfs  rfc KsDttddddgdddddtd dd d d }tdd|it|f|S)z ResNeSt-269e model. Matches paper ResNeSt-269 model, https://arxiv.org/abs/2004.08955 Since this codebase supports all possible variations, 'e' for deep stem, stem_width 64, avg in downsample. r0rOrUrTrrWFrXrY resnest269er?)rir\r^r/r/r0ris  ric KsDttddddgdddddtdddd d }td d |it|f|S)z]ResNeSt-50 4s2x40d from https://github.com/zhanghang1989/ResNeSt/blob/master/ablation.md rr r`rUrVT(rWrXrYresnest50d_4s2x40dr?)rkr\r^r/r/r0rks  rkc KsDttddddgdddddtdddd d }td d |it|f|S)z]ResNeSt-50 1s4x24d from https://github.com/zhanghang1989/ResNeSt/blob/master/ablation.md rr r`rUrVTrdrrXrYresnest50d_1s4x24dr?)rlr\r^r/r/r0rls  rlN)F)rB)F)F)F)F)F)F)F)F)r<ZtorchrZ timm.datarrZ timm.layersrZ_builderr _registryr r Zresnetr Moduler rArLZ default_cfgsr[r_rarcrfrirkrlr/r/r/r0sr    f