a PSic.B@sddlZddlmZddlmZmZmZmZmZddl Z ddl m Z m Z ddl m Z mZddlmZddlmZd d lmZmZd d lmZd d lmZmZmZgd ZGdddeZGdddZGddde jZ Gddde jZ!d/e"e#e$e$edddZ%eee&eee$ee!dddZ'dedZ(Gd d!d!eZ)Gd"d#d#eZ*ed$e)j+fd%dd&d'ee)e$ee!d(d)d*Z,ed$e*j+fd%dd&d'ee*e$ee!d(d+d,Z-d d-lm.Z.e.e)j+j/e*j+j/d.Z0dS)0N)partial)AnyCallableListOptionalSequence)nnTensor)Conv2dNormActivationSqueezeExcitation)ImageClassification)_log_api_usage_once) WeightsEnumWeights)_IMAGENET_CATEGORIES)handle_legacy_interface_ovewrite_named_param_make_divisible) MobileNetV3MobileNet_V3_Large_WeightsMobileNet_V3_Small_Weightsmobilenet_v3_largemobilenet_v3_smallcs*eZdZdZdeedfdd ZZS)r Z DEPRECATED)input_channelssqueeze_factorcsDt||d}tj||tjd|j|_t|dt dt dS)Nscale_activation activationzThis SqueezeExcitation class is deprecated since 0.12 and will be removed in 0.14. Use torchvision.ops.SqueezeExcitation instead.) rsuper__init__r Hardsigmoidr!reludelattrwarningswarn FutureWarning)selfrrsqueeze_channels __class__Z/var/www/html/django/DPS/env/lib/python3.9/site-packages/torchvision/models/mobilenetv3.pyr#s zSqueezeExcitation.__init__)r)__name__ __module__ __qualname____doc__intr# __classcell__r.r.r,r/r sr c @s>eZdZeeeeeeeeed ddZeeedddZ dS)InvertedResidualConfig) rkernelexpanded_channels out_channelsuse_ser!stridedilation width_multc CsP||| |_||_||| |_||| |_||_|dk|_||_||_dS)NHS) adjust_channelsrr7r8r9r:use_hsr;r<) r*rr7r8r9r:r!r;r<r=r.r.r/r#*s  zInvertedResidualConfig.__init__channelsr=cCst||dS)Nr)rrAr.r.r/r??sz&InvertedResidualConfig.adjust_channelsN) r0r1r2r4boolstrfloatr# staticmethodr?r.r.r.r/r6(s r6csXeZdZeeejdfeedej fedej fdfdd Z e e dddZ Z S) InvertedResidualr.)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_sizerIactivation_layer)rLr;r<groupsrIrMrr)r"r#r; ValueErrorrr9use_res_connectr@r HardswishReLUr8appendr r<r7r:r Sequentialblock_is_cn)r*rHrIrJlayersrMr;r+r,r.r/r#FsP     zInvertedResidual.__init__)inputreturncCs||}|jr||7}|SN)rUrP)r*rXresultr.r.r/forward~s zInvertedResidual.forward)r0r1r2rSElayerrr$r6rModuler#r r\r5r.r.r,r/rGDs   8rGc sreZdZd eeeeeedej feedej fe e ddfdd Z e e dd d Ze e dd d ZZS)rN皙?.)inverted_residual_setting last_channel num_classesrUrIdropoutkwargsrYc stt||s tdn$t|trz(MobileNetV3.__init__..zDThe inverted_residual_setting should be List[InvertedResidualConfig]NgMbP?g{Gz?)epsmomentumrr )rLr;rIrMrrKT)inplace)prpfan_out)mode)"r"r#rrOrfrall TypeErrorrGrr BatchNorm2drrSr rQr9rTfeaturesAdaptiveAvgPool2davgpoolLinearDropout classifiermodulesConv2dinitkaiming_normal_weightbiaszeros_ GroupNormones_normal_)r*rarbrcrUrIrdrerWfirstconv_output_channelsrHlastconv_input_channelslastconv_output_channelsmr,r.r/r#sr             zMobileNetV3.__init__)xrYcCs.||}||}t|d}||}|S)Nr)rwrytorchflattenr|r*rr.r.r/ _forward_impls     zMobileNetV3._forward_implcCs ||SrZ)rrr.r.r/r\szMobileNetV3.forward)r_NNr`)r0r1r2rr6r4rrrr^rErr#r rr\r5r.r.r,r/rs [ r?F)archr= reduced_taildilatedrec 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=rrmFRE@H(TxPr>ipiiirX`0i i@izUnsupported model type )rr6r?rO) rr=rrreZreduce_dividerr< bneck_confr?rarbr.r.r/_mobilenet_v3_confsL          r)rarbweightsprogressrerYcKsN|durt|dt|jdt||fi|}|durJ||j|d|S)Nrc categories)r)rlenmetarload_state_dictget_state_dict)rarbrrremodelr.r.r/ _mobilenet_v3 s r)rr)min_sizerc @sveZdZedeeddiedddddd id d d Zed eedddiedddddd idd d ZeZ 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@5zJThese weights were trained from scratch by using a simple training recipe. num_paramsrecipe_metrics_docsurl transformsrzChttps://download.pytorch.org/models/mobilenet_v3_large-5c1a4163.pth)r resize_sizezHhttps://github.com/pytorch/vision/issues/3995#new-recipe-with-reg-tuninggK7R@gNbX9$W@a/ These weights improve marginally upon the results of the original paper by using a modified version of TorchVision's `new training recipe `_. N) r0r1r2rrr _COMMON_META IMAGENET1K_V1 IMAGENET1K_V2DEFAULTr.r.r.r/r8sB  rc @sBeZdZedeeddiedddddd id d d ZeZd S)rzChttps://download.pytorch.org/models/mobilenet_v3_small-047dcff4.pthrri&rrgnP@g}?5^U@rz} These weights improve upon the results of the original paper by using a simple training recipe. rrN) r0r1r2rrr rrrr.r.r.r/r`s" r pretrained)rT)rr)rrrerYcKs2t|}tdi|\}}t||||fi|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.resnet.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)rverifyrrrrrerarbr.r.r/rvs rcKs2t|}tdi|\}}t||||fi|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.resnet.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 r) _ModelURLs)rr)rFF)1r' functoolsrtypingrrrrrrrr Zops.miscr r r]Ztransforms._presetsr utilsr_apirr_metar_utilsrrr__all__r6r^rGrrDrErCrr4rrrrrrrrr model_urlsr.r.r.r/sb     Ak 2 (