Ng< ~dZddlmZddlmZddlmZGddejZeZGdd ejZ d S) a  Depthwise Separable Conv Modules Basic DWS convs. Other variations of DWS exist with batch norm or activations between the DW and PW convs such as the Depthwise modules in MobileNetV2 / EfficientNet and Xception. Hacked together by / Copyright 2020 Ross Wightman )nn) create_conv2d)get_norm_act_layerc eZdZdZdddddddejejddf fd Zed Z ed Z d Z xZ S) SeparableConvNormActz4 Separable Conv w/ trailing Norm and Activation rF?TNc ftt|t|t ||z||||d|_tt ||z|| |||_t| | }| t| ni}||fd| i||_ dS)NTstridedilationpadding depthwiserbias) drop_layer apply_act) superr__init__rintconv_dwconv_pwrdictbn)self in_channels out_channels kernel_sizerrrrchannel_multiplierpw_kernel_size norm_layer act_layerrrnorm_act_layer norm_kwargs __class__s V/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/layers/separable_conv.pyrzSeparableConvNormAct.__init__s "D))22444$ [+==>> HgOOO %  00 1 1<Y`gkmmm ,J BB5?5Kdj1111QS  .RRRkRRc|jjSNrrrs r(rz SeparableConvNormAct.in_channels! |''r)c|jjSr+rrr-s r(rz!SeparableConvNormAct.out_channels% |((r)c||}||}||}|Sr+)rrrrxs r(forwardzSeparableConvNormAct.forward)s4 LLOO LLOO GGAJJr)) __name__ __module__ __qualname____doc__r BatchNorm2dReLUrpropertyrrr5 __classcell__r's@r(rrs>?TU_ahm$'bn`b`gDSSSSSS ((X())X)r)rcZeZdZdZ d fd ZedZed Zd ZxZ S) SeparableConv2dz Separable Conv r rr Fr c tt|t|t ||z||||d|_tt ||z|| |||_dS)NTr r)rr@rrrrr) rrrr rrrrr!r"r's r(rzSeparableConv2d.__init__6s ot$$--///$ [+==>> HgOOO %  00 1 1<Y`gkmmm r)c|jjSr+r,r-s r(rzSeparableConv2d.in_channelsAr.r)c|jjSr+r0r-s r(rzSeparableConv2d.out_channelsEr1r)cZ||}||}|Sr+)rrr3s r(r5zSeparableConv2d.forwardIs% LLOO LLOOr))r rrr Fr r) r6r7r8r9rr<rrr5r=r>s@r(r@r@3shm89 m m m m m m((X())X)r)r@N) r9torchrrcreate_norm_actrModulerSeparableConvBnActr@r)r(rJs((((((//////29D*bir)