|j$ddlmZddlmZmZddlmZmZddlZddlm Z ddl m Z ddl m Z erdd lmZdd lmZGd d eZgZd diZGdde jZGdde jZ d d!dZ d"d#dZdS)$) annotations) TYPE_CHECKING TypedDict) NotRequiredUnpackN)nn)get_weights_path_from_url)ConvNormActivation)Tensor)Size2c$eZdZUded<ded<dS)_MobileNetV1OptionszNotRequired[int] num_classeszNotRequired[bool] with_poolN)__name__ __module__ __qualname____annotations__p/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/paddle/vision/models/mobilenetv1.pyrr"s*%%%%$$$$$$rrzmobilenetv1_1.0)zAhttps://paddle-hapi.bj.bcebos.com/models/mobilenetv1_1.0.pdparams 3033ab1975b1670bef51545feb65fc45c(eZdZdfd ZddZxZS)DepthwiseSeparable in_channelsint out_channels1 out_channels2 num_groupsstrider scalefloatreturnNonec .tt|t||zd|dt||z|_tt||zt||zddd|_dS)N) kernel_sizer!paddinggroupsr)r)r!r*)super__init__r r_depthwise_conv_pointwise_conv)selfrrrr r!r" __class__s rr-zDepthwiseSeparable.__init__3s 1   % & &zE)**     2  % & &  % & &    rxr cZ||}||}|S)N)r.r/)r0r2s rforwardzDepthwiseSeparable.forwardOs-   # #   # #r)rrrrrrr rr!r r"r#r$r%r2r r$r )rrrr-r4 __classcell__r1s@rrr2sQ      8rrcTeZdZUdZded<ded<ded< ddfd ZddZxZS) MobileNetV1aMobileNetV1 model from `"MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications" `_. 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 MobileNetV1 model. Examples: .. code-block:: pycon >>> import paddle >>> from paddle.vision.models import MobileNetV1 >>> model = MobileNetV1() >>> x = paddle.rand([1, 3, 224, 224]) >>> out = model(x) >>> print(out.shape) paddle.Size([1, 1000]) r#r"rrboolr?Tr$r%c t||_g|_||_||_t dtd|zddd|_| ttd|zdddd|d }|j || ttd|zdd dd|d }|j || ttd |zd d d d|d }|j || ttd |zd d d d|d }|j || ttd |zd d d d|d }|j || ttd |zd dd d|d } |j | tdD]i} | ttd|zdddd|dt| dzz } |j | j| ttd|zdddd|d } |j | | ttd|zdddd|d } |j | |rtjd|_|dkr,tjtd|z||_dSdS)Nr' r r()r out_channelsr)r!r*@)rrrr r!r"conv2_1)sublayernameconv2_2conv3_1conv3_2conv4_1iconv4_2conv5_iconv5_6conv6r)r,r-r"dwslrrr rconv1 add_sublayerrappendrangestrrAdaptiveAvgPool2D pool2d_avgLinearfc)r0r"rrdws21dws22dws31dws32dws41dws42itmpdws56dws6r1s rr-zMobileNetV1.__init__usx   &"'R%Z     !!'U OO   "    !!'U OO ! "    !!'e ,,!! "    !!'e ,,!! "    !!'e ,,!! "    !!'e ,,!! "    q " "A##+ #C%K 0 0"%"%" AE *$  C I  S ! ! ! !!!'e ,,!" "      'u --"" !      6 2155DO ??iD5L 1 1;??DGGG ?rr2r c||}|jD] }||}|jr||}|jdkr*t j|d}||}|S)Nrr()rPrOrrVrpaddleflattenrX)r0r2dwss rr4zMobileNetV1.forwards JJqMM9  CAAA > #""A  a  q!$$A Ar)r;r<T)r"r#rrrr:r$r%r5)rrr__doc__rr-r4r6r7s@rr9r9Us6LLLOOO N@N@N@N@N@N@N@`        rr9FarchrT pretrainedr:kwargsUnpack[_MobileNetV1Options]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)r9 model_urlsr rdload load_dict)rhrirjmodel weight_pathparams r _mobilenetrss  ! !& ! !E z!!! _ _ _"!!0 t Q D!1!!4   K((  Lrr;r"r#c Ftdt|z|fd|i|}|S)aMobileNetV1 from `"MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications" `_. 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:`MobileNetV1 `. Returns: :ref:`api_paddle_nn_Layer`. An instance of MobileNetV1 model. Examples: .. code-block:: pycon >>> import paddle >>> from paddle.vision.models import mobilenet_v1 >>> # Build model >>> model = mobilenet_v1() >>> # Build model and load imagenet pretrained weight >>> # model = mobilenet_v1(pretrained=True) >>> # build mobilenet v1 with scale=0.5 >>> model_scale = mobilenet_v1(scale=0.5) >>> x = paddle.rand([1, 3, 224, 224]) >>> out = model(x) >>> print(out.shape) paddle.Size([1, 1000]) mobilenetv1_r")rsrT)rir"rjrps r mobilenet_v1rv%s@L U#Z  7< @F  E Lr)F)rhrTrir:rjrkr$r9)Fr;)rir:r"r#rjrkr$r9) __future__rtypingrrtyping_extensionsrrrdrpaddle.utils.downloadr opsr r paddle._typingr r__all__rmLayerrr9rsrvrrrrs#""""" 21111111 ;;;;;;$$$$$$%$$$$$$%%%%%i%%%          F{{{{{"({{{~#(&)))))))r