U ʹju<@s,dZddlmZddlZddlmZddlmZmZddl m Z m Z m Z m Z mZmZmZddlmZddlmZdd lmZmZd gZGd d d ejZGd ddejZGdddejZGdd d ejZeedddgddddgddddgdddddgddddedddgddddgddddgdddd d!gddddedddgdd"d#d$gd$dd%dgd!ddddgd&d&d'dedddgddddgddddgd!ddddgd&d&d'dedddgdd"d#d$gd$dd%dgd!ddddgd&dd'dedddgddddgddddgd!ddddgd&dd'dedddgddddgddddgdddddgd&dd'dedddgddddgddddgdddd d!gd&dd'dedddgddddgddddgddd!d(d!gd&dd'dedddgddddgddddgdddddgd&dd)dd* Z e d+e d,<dJd-d.Z!dKd/d0Z"ee"dd1e"dd1e"dd1e"d2d3d4d5e"dd1e"d2d3d4d5e"dd1e"dd1e"dd1e"dd1d6 Z#edLed7d8d9Z$edMed7d:d;Z%edNed7dd?Z'edPed7d@dAZ(edQed7dBd+Z)edRed7dCdDZ*edSed7dEdFZ+edTed7dGdHZ,edUed7dId,Z-dS)Va VoVNet (V1 & V2) Papers: * `An Energy and GPU-Computation Efficient Backbone Network` - https://arxiv.org/abs/1904.09730 * `CenterMask : Real-Time Anchor-Free Instance Segmentation` - https://arxiv.org/abs/1911.06667 Looked at https://github.com/youngwanLEE/vovnet-detectron2 & https://github.com/stigma0617/VoVNet.pytorch/blob/master/models_vovnet/vovnet.py for some reference, rewrote most of the code. Hacked together by / Copyright 2020 Ross Wightman )ListNIMAGENET_DEFAULT_MEANIMAGENET_DEFAULT_STD) ConvNormActSeparableConvNormActBatchNormAct2dClassifierHeadDropPath create_attncreate_norm_act_layer)build_model_with_cfg)checkpoint_seq)register_modelgenerate_default_cfgsVovNetcs8eZdZfddZejeejejdddZZS)SequentialAppendListcstt|j|dSN)superr__init__)selfargs __class__Q/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/models/vovnet.pyrszSequentialAppendList.__init__)x concat_listreturncCsNt|D]2\}}|dkr(|||q|||dqtj|dd}|S)Nrr )dim) enumerateappendtorchcat)rrrimodulerrrforward!s zSequentialAppendList.forward) __name__ __module__ __qualname__rr$ZTensorrr( __classcell__rrrrrs rcs4eZdZdddeejdffdd ZddZZS)OsaBlockFNc stt|||_||_t|| d} |} |jrT| |krT|r@tt| |df| |_nd|_g} t |D]8}|jrt ||f| }nt| |df| }|} | |qft | |_ |||} t| |f| |_|rt||nd|_| |_dS)N norm_layer act_layerr )rr-rresidual depthwisedictAssertionErrorrconv_reductionrangerr#rconv_mid conv_concatr attn drop_path)rin_chsmid_chsout_chslayer_per_blockr3r4r;r0r1r< conv_kwargsZ next_in_chsZ mid_convsr&convrrrr-s*      zOsaBlock.__init__cCsn|g}|jdk r||}|||}||}|jdk rD||}|jdk rX||}|jrj||d}|S)Nr)r7r9r:r;r<r3)rroutputrrrr(Ys         zOsaBlock.forward r)r*r+rnnReLUrr(r,rrrrr-+s,r-cs6eZdZddddeejdffdd ZddZZS) OsaStageTFeseNc stt|d|_|r,tjdddd|_nd|_g} t|D]l}||dk}| dk rp| |dkrpt| |}nd}| t |||||o|dk||r| nd | | |d g7} |}q>tj | |_ dS) NFr2T)Z kernel_sizestrideZ ceil_moder rr.)r3r4r;r0r1r<) rrGrgrad_checkpointingrEZ MaxPool2dpoolr8r r- Sequentialblocks)rr=r>r?block_per_stager@ downsampler3r4r;r0r1drop_path_ratesrOr&Z last_blockr<rrrrjs2     zOsaStage.__init__cCs@|jdk r||}|jr2tjs2t|j|}n ||}|Sr)rMrLr$jitZ is_scriptingrrOrrrrrr(s    zOsaStage.forwardrDrrrrrGhs %rGcseZdZddddeejddffdd Zejj dd d Z ejj dd d Z ejj ddZ dddZ ddZdedddZddZZS)rr2avg rKc  stt|||_||_|dks&tt|f| }|dd} |d} |d} |d}|d}|d}t||d }| d }|d rtnt }t j t || d d fdd i||| d | dd fddi||| d| d d fd|i|g|_ t| dd d| dkrdnd dg|_ | }ttd | t||}| dd|dd}tf|d|d |dd|}g}tdD]}| d kp|d k}|t||| ||||||f|||d|g7}|||_||rd nd9}|j t|j|d|dg7_ qzt j ||_t|j|||d|_|D]J\}}t|t jr^t jj|jdddnt|t jr2t j|j q2dS)a Args: cfg (dict): Model architecture configuration in_chans (int): Number of input channels (default: 3) num_classes (int): Number of classifier classes (default: 1000) global_pool (str): Global pooling type (default: 'avg') output_stride (int): Output stride of network, one of (8, 16, 32) (default: 32) norm_layer (Union[str, nn.Module]): normalization layer act_layer (Union[str, nn.Module]): activation layer drop_rate (float): Dropout rate (default: 0.) drop_path_rate (float): Stochastic depth drop-path rate (default: 0.) kwargs (dict): Extra kwargs overlayed onto cfg rW stem_stridestem_chsstage_conv_chs stage_out_chsrPr@r/rIr4rr2rJr zstem.)Znum_chsZ reductionr'r Nr3r;)r3r4r;)rQrRzstages.Z pool_type drop_rateZfan_outZrelu)modeZ nonlinearity)!rrr num_classesr^r6r5getrrrErNstemZ feature_infor$splitZlinspacesumr8rG num_featuresstagesr headZ named_modules isinstanceZConv2dinitZkaiming_normal_weightZLinearZzeros_Zbias)rcfgZin_chansr` global_poolZ output_strider0r1r^Zdrop_path_ratekwargsrXrZr[r\rPr@rAZlast_stem_strideZ conv_typeZcurrent_strideZ stage_dprZ in_ch_listZ stage_argsrfr&rQnmrrrrsj     "  & zVovNet.__init__FcCstd|r dnddS)Nz^stemz^stages\.(\d+)z^stages\.(\d+).blocks\.(\d+))rbrO)r5)rZcoarserrr group_matchers zVovNet.group_matcherTcCs|jD] }||_qdSr)rfrL)renablesrrrset_grad_checkpointings zVovNet.set_grad_checkpointingcCs|jjSr)rgfc)rrrrget_classifierszVovNet.get_classifiercCst|j|||jd|_dS)Nr])r rer^rg)rr`rlrrrreset_classifierszVovNet.reset_classifiercCs||}||Sr)rbrfrTrrrforward_featuress zVovNet.forward_features pre_logitscCs|j||dS)Nrx)rg)rrryrrr forward_headszVovNet.forward_headcCs||}||}|Sr)rwrzrTrrrr(s  zVovNet.forward)F)T)rV)F)r)r*r+rrErFrr$rSignorerprsrurvrwboolrzr(r,rrrrrs&S    @iiirIFr.)rZr[r\r@rPr3r4r;rYr2P`piTrH Zeca) vovnet39a vovnet57aese_vovnet19b_slim_dwese_vovnet19b_dwese_vovnet19b_slimZ ese_vovnet19b ese_vovnet39b ese_vovnet57b ese_vovnet99b eca_vovnet39brese_vovnet39b_evoscKs$tt||ft|tddd|S)NT)Zflatten_sequential)Z model_cfgZ feature_cfg)rr model_cfgsr5)variant pretrainedrmrrr_create_vovnet}src Ks|dddddttddd |S) NrU)r2rr)rg?Zbicubicz stem.0.convzhead.fc) urlr`Z input_sizeZ pool_sizeZcrop_pct interpolationZmeanZstdZ first_conv classifierr)rrmrrr_cfgsr)rztimm/)r2 rgffffff?)Z hf_hub_idZtest_input_sizeZ test_crop_pct) zvovnet39a.untrainedzvovnet57a.untrainedzese_vovnet19b_slim_dw.untrainedzese_vovnet19b_dw.ra_in1kzese_vovnet19b_slim.untrainedzese_vovnet39b.ra_in1kzese_vovnet57b.untrainedzese_vovnet99b.untrainedzeca_vovnet39b.untrainedzese_vovnet39b_evos.untrained)rcKstdd|i|S)Nrr)rrrrmrrrrsrcKstdd|i|S)Nrr)rrrrrrrsrcKstdd|i|S)Nrr)rrrrrrrsrcKstdd|i|S)Nrr)rrrrrrrsrcKstdd|i|S)Nrr)rrrrrrrsrcKstdd|i|S)Nrr)rrrrrrrscKstdd|i|S)Nrr)rrrrrrrsrcKstdd|i|S)Nrr)rrrrrrrsrcKstdd|i|S)Nrr)rrrrrrrsrcKsdd}td||d|S)Nc[std|fddi|S)NZ evonorms0rSF)r )renkwargsrrr norm_act_fnsz'ese_vovnet39b_evos..norm_act_fnr)rr0)rr)rrmrrrrrs)F)r.)F)F)F)F)F)F)F)F)F)F).__doc__typingrr$Ztorch.nnrEZ timm.datarrZ timm.layersrrrr r r r Z_builderrZ _manipulater _registryrr__all__rNrModuler-rGrr5rrrZ default_cfgsrrrrrrrrrrrrrrs<  $  =1w                                       j