U h?@sddlmZddlmZmZmZmZmZddlZddlm Z m Z ddl m Z m ZddlmZddlmZd d lmZmZmZd d lmZd d lmZmZmZd ddddgZGdddZGddde jZ Gdd d e jZ!d*e"e#e$e$edddZ%eee&eee$ee!dddZ'ded Z(Gd!ddeZ)Gd"ddeZ*eed#e)j+fd$dd%d&ee)e$ee!d'd(dZ,eed#e*j+fd$dd%d&ee*e$ee!d'd)dZ-dS)+)partial)AnyCallableListOptionalSequenceN)nnTensor)Conv2dNormActivationSqueezeExcitation)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_make_divisible_ovewrite_named_paramhandle_legacy_interface MobileNetV3MobileNet_V3_Large_WeightsMobileNet_V3_Small_Weightsmobilenet_v3_largemobilenet_v3_smallc @s>eZdZeeeeeeeeed ddZeeedddZ dS)InvertedResidualConfig) input_channelskernelexpanded_channels out_channelsuse_se activationstridedilation width_multc CsP||| |_||_||| |_||| |_||_|dk|_||_||_dS)NHS) adjust_channelsrrrr r!use_hsr#r$) selfrrrr r!r"r#r$r%r*P/var/www/html/venv/lib/python3.8/site-packages/torchvision/models/mobilenetv3.py__init__s  zInvertedResidualConfig.__init__Zchannelsr%cCst||dS)N)rr-r*r*r+r'/sz&InvertedResidualConfig.adjust_channelsN) __name__ __module__ __qualname__intboolstrfloatr, staticmethodr'r*r*r*r+rs rcsXeZdZeeejdfeedej fedej fdfdd Z e e dddZ Z S) InvertedResidual)Zscale_activation.)cnf norm_layerse_layerc s$td|jkr dks*ntd|jdko>|j|jk|_g}|jrRtj ntj }|j |jkr| t |j|j d||d|jdkrdn|j}| t |j |j |j||j|j ||d|jrt|j dd}| ||j || t |j |jd|ddtj||_|j|_|jdk|_dS)Nrr zillegal stride value kernel_sizer9activation_layer)r<r#r$groupsr9r=r.)superr,r# ValueErrorrr use_res_connectr(r HardswishZReLUrappendr r$rr!r SequentialblockZ_is_cn)r)r8r9r:layersr=r#Zsqueeze_channels __class__r*r+r,6sX     zInvertedResidual.__init__)inputreturncCs||}|jr||7}|SN)rFrB)r)rJresultr*r*r+forwardns zInvertedResidual.forward)r/r0r1rSElayerrZ HardsigmoidrrModuler,r rN __classcell__r*r*rHr+r74s   8r7c sreZdZd eeeeeedej feedej fe e ddfdd Z e e dd d Ze e dd d ZZS)rN皙?.)inverted_residual_setting last_channel num_classesrFr9dropoutkwargsrKc stt||s tdn$t|trsz(MobileNetV3.__init__..zDThe inverted_residual_setting should be List[InvertedResidualConfig]NgMbP?g{Gz?)epsZmomentumrr )r<r#r9r=rr;T)inplace)praZfan_out)mode)"r@r,rrArYrall TypeErrorr7rrZ BatchNorm2drrDr rCr rEfeaturesZAdaptiveAvgPool2davgpoolZLinearZDropout classifiermodulesZConv2dinitZkaiming_normal_weightZbiasZzeros_Z GroupNormZones_Znormal_)r)rTrUrVrFr9rWrXrGZfirstconv_output_channelsr8Zlastconv_input_channelsZlastconv_output_channelsmrHr*r+r,vsr             zMobileNetV3.__init__)xrKcCs.||}||}t|d}||}|S)Nr)rfrgtorchflattenrhr)rmr*r*r+ _forward_impls     zMobileNetV3._forward_implcCs ||SrL)rqrpr*r*r+rNszMobileNetV3.forward)rRNNrS)r/r0r1rrr2rrrrPr5rr,r rqrNrQr*r*rHr+rus [ ?F)archr% reduced_taildilatedrXc Ks|rdnd}|rdnd}tt|d}ttj|d}|dkr|dddddddd|ddd d dddd|d dd d dddd|d d d d dddd|d d dd dddd|d d dd dddd|d ddddddd|dddddddd|dddddddd|dddddddd|dddddddd|dddddddd|dd dd|ddd||d|d d|d|ddd||d|d d|d|ddd|g} |d|} n"|dkr|dddddddd|ddd d dddd|d ddd dddd|d d dd dddd|d d dd dddd|d d dd dddd|d d dddddd|dd dddddd|dd d d|ddd||d|d d!|d|ddd||d|d d!|d|ddd|g } |d"|} ntd#|| | fS)$Nr r)r%rr^FZRE@H(TxPr&ipiiirX`0i i@izUnsupported model type )rrr'rA) rsr%rtrurXZreduce_dividerr$Z bneck_confr'rTrUr*r*r+_mobilenet_v3_confsL          r)rTrUweightsprogressrXrKcKsL|dk rt|dt|jdt||f|}|dk rH||j|dd|S)NrV categoriesT)rZ check_hash)rlenmetarZload_state_dictZget_state_dict)rTrUrrrXmodelr*r*r+ _mobilenet_v3s r)rr)Zmin_sizerc @sveZdZedeeddedddddd id d d d dZedeedddedddddd id ddd dZeZ dS)rzChttps://download.pytorch.org/models/mobilenet_v3_large-8738ca79.pth crop_sizeiS^https://github.com/pytorch/vision/tree/main/references/classification#mobilenetv3-large--small ImageNet-1Kg R@g(\V@zacc@1zacc@5g-?gw/5@zJThese weights were trained from scratch by using a simple training recipe.Z num_paramsZrecipeZ_metricsZ_ops _file_sizeZ_docsurlZ transformsrzChttps://download.pytorch.org/models/mobilenet_v3_large-5c1a4163.pth)rZ resize_sizezHhttps://github.com/pytorch/vision/issues/3995#new-recipe-with-reg-tuninggK7R@gNbX9$W@gZd5@a/ These weights improve marginally upon the results of the original paper by using a modified version of TorchVision's `new training recipe `_. N) r/r0r1rrr _COMMON_META IMAGENET1K_V1Z IMAGENET1K_V2DEFAULTr*r*r*r+r(sB  c @sBeZdZedeeddedddddd id d d d dZeZdS)rzChttps://download.pytorch.org/models/mobilenet_v3_small-047dcff4.pthrri&rrgnP@g}?5^U@rgv/?g r#@z} These weights improve upon the results of the original paper by using a simple training recipe. rrN) r/r0r1rrr rrrr*r*r*r+rTs" Z pretrained)rT)rr)rrrXrKcKs*t|}td|\}}t||||f|S)a Constructs a large MobileNetV3 architecture from `Searching for MobileNetV3 `__. Args: weights (:class:`~torchvision.models.MobileNet_V3_Large_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.MobileNet_V3_Large_Weights` below for more details, and possible values. By default, no pre-trained weights are used. progress (bool, optional): If True, displays a progress bar of the download to stderr. Default is True. **kwargs: parameters passed to the ``torchvision.models.mobilenet.MobileNetV3`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.MobileNet_V3_Large_Weights :members: r)r)rverifyrrrrrXrTrUr*r*r+rls cKs*t|}td|\}}t||||f|S)a Constructs a small MobileNetV3 architecture from `Searching for MobileNetV3 `__. Args: weights (:class:`~torchvision.models.MobileNet_V3_Small_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.MobileNet_V3_Small_Weights` below for more details, and possible values. By default, no pre-trained weights are used. progress (bool, optional): If True, displays a progress bar of the download to stderr. Default is True. **kwargs: parameters passed to the ``torchvision.models.mobilenet.MobileNetV3`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.MobileNet_V3_Small_Weights :members: r)r)rrrrrr*r*r+rs )rrFF). functoolsrtypingrrrrrrnrr Zops.miscr r rOZtransforms._presetsr utilsrZ_apirrr_metarZ_utilsrrr__all__rrPr7rr4r5r3rr2rrrrrrrr*r*r*r+sx     Ak 2 ,