{-jZ ddlmZddlmZmZmZddlmZmZddl Z ddl m Z ddl m Z ddl mZd d lmZerdd l mZGd d eZgZddiZGdde jZGdde jZ d!d"dZ d#d$d ZdS)%) annotations) TYPE_CHECKINGCallable TypedDict) NotRequiredUnpackN)nn)get_weights_path_from_url)ConvNormActivation)_make_divisible)Tensorc$eZdZUded<ded<dS)_MobileNetV2OptionszNotRequired[int] num_classeszNotRequired[bool] with_poolN)__name__ __module__ __qualname____annotations__`/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/vision/models/mobilenetv2.pyrr#s*%%%%$$$$$$rrzmobilenetv2_1.0)zChttps://paddle-hapi.bj.bcebos.com/models/mobilenet_v2_x1.0.pdparams 0340af0a901346c8d46f4529882fb63dc6eZdZejfdfd ZddZxZS)InvertedResidualinpintoupstride expand_ratiofloat norm_layerCallable[..., nn.Layer]returnNonec t||_|dvsJtt ||z}|jdko||k|_g}|dkr1|t||d|tj | t|||||tj tj ||dddd||gtj ||_ dS)N)r r r  kernel_sizer$activation_layer)r!groupsr$r+rF) bias_attr)super__init__r!rrounduse_res_connectappendr r ReLU6extendConv2D Sequentialconv) selfrr r!r"r$ hidden_dimlayers __class__s rr/zInvertedResidual.__init__3s,  s\12233 #{a/>C3J 1   MM" !)%'X      "!%)%'X  *c1aeDDD 3  M6* rxrcj|jr|||zS||S)N)r1r7r8r<s rforwardzInvertedResidual.forward]s2   tyy||# #99Q<< r) rrr rr!rr"r#r$r%r&r'r<rr&r)rrrr BatchNorm2Dr/r? __classcell__r;s@rrr2s^/1n (+(+(+(+(+(+(+T        rrcJeZdZUdZded<ded< ddfd ZddZxZS) MobileNetV2agMobileNetV2 model from `"MobileNetV2: Inverted Residuals and Linear Bottlenecks" `_. Args: scale (float, optional): Scale of channels in each layer. Default: 1.0. num_classes (int, optional): Output dim of last fc layer. If num_classes <= 0, last fc layer will not be defined. Default: 1000. with_pool (bool, optional): Use pool before the last fc layer or not. Default: True. Returns: :ref:`api_paddle_nn_Layer`. An instance of MobileNetV2 model. Examples: .. code-block:: pycon >>> import paddle >>> from paddle.vision.models import MobileNetV2 >>> model = MobileNetV2() >>> x = paddle.rand([1, 3, 224, 224]) >>> out = model(x) >>> print(out.shape) paddle.Size([1, 1000]) rrboolr?Tscaler#r&r'c t||_||_d}d}t}d}t j}gdgdgdgdgdgd gd g} t||z|}t|td |z||_ td |d |t j g} | D][\} } } }t| |z|}t| D]1}|dkr|nd}| ||||| ||}2\| t||j d|t j t j| |_|rt jd|_|jdkrFt jt jdt j|j ||_dSdS)N i)r r r )r r )rNrKr )rN@r )rN`rPr )rNrPr )rNi@r r rGrPr )r!r$r+rr )r"r$r)g?)r.r/rrrr rArmax last_channelr r3ranger2r6featuresAdaptiveAvgPool2D pool2d_avgDropoutLinear classifier)r8rIrr input_channelrVblock round_nearestr$inverted_residual_settingrXtcnsoutput_channelir!r;s rr/zMobileNetV2.__init__s. &"    ^ MM MM MM MM MM NN NN% !( (=}MM + 3sE?? *M   %!#     4 / /JAq!Q,QY FFN1XX / /1ff!E%&%&#- !/  /  !%!#        x0  6 2155DO  a   m 34+s rr?zMobileNetV2.forwardsc MM!   > #""A  a  q!$$A""Ar)rGrHT)rIr#rrrrFr&r'r@)rrr__doc__rr/r?rBrCs@rrErEds6OOO HHHHHHHT        rrEFarchstr pretrainedrFkwargsUnpack[_MobileNetV2Options]r&c tdi|}|rq|tvs J|dtt|dt|d}tj|}|||S)NzJ model do not have a pretrained model now, you should set pretrained=Falserr r)rE model_urlsr riload load_dict)rlrnromodel weight_pathparams r _mobilenetrxs  ! !& ! !E z!!! _ _ _"!!0 t Q D!1!!4   K((  LrrGrIr#c Ftdt|z|fd|i|}|S)aMobileNetV2 from `"MobileNetV2: Inverted Residuals and Linear Bottlenecks" `_. Args: pretrained (bool, optional): Whether to load pre-trained weights. If True, returns a model pre-trained on ImageNet. Default: False. scale (float, optional): Scale of channels in each layer. Default: 1.0. **kwargs (optional): Additional keyword arguments. For details, please refer to :ref:`MobileNetV2 `. Returns: :ref:`api_paddle_nn_Layer`. An instance of MobileNetV2 model. Examples: .. code-block:: pycon >>> import paddle >>> from paddle.vision.models import mobilenet_v2 >>> # Build model >>> model = mobilenet_v2() >>> # Build model and load imagenet pretrained weight >>> # model = mobilenet_v2(pretrained=True) >>> # Build mobilenet v2 with scale=0.5 >>> model = mobilenet_v2(scale=0.5) >>> x = paddle.rand([1, 3, 224, 224]) >>> out = model(x) >>> print(out.shape) paddle.Size([1, 1000]) mobilenetv2_rI)rxrm)rnrIrorus r mobilenet_v2r{s@J U#Z  7< @F  E Lr)F)rlrmrnrFrorpr&rE)FrG)rnrFrIr#rorpr&rE) __future__rtypingrrrtyping_extensionsrrrir paddle.utils.downloadr opsr _utilsrrr__all__rrLayerrrErxr{rrrrs#""""" 21111111 ;;;;;;$$$$$$######%%%%%%i%%%  / / / / / rx/ / / drrrrr"(rrrl#(&(((((((r