2VhX-dZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd l mZdd lmZdd lmZGd de Zy)z2Keras base class for depthwise convolution layers.) activations) constraints) initializers)ops) regularizers)standardize_data_format) InputSpec)Layer)compute_conv_output_shape)standardize_padding)standardize_tuplecdeZdZdZ dfd ZdZdZdZdZfdZ xZ S) BaseDepthwiseConvaAbstract N-D depthwise convolution layer. Depthwise convolution is a type of convolution in which each input channel is convolved with a different kernel (called a depthwise kernel). You can understand depthwise convolution as the first step in a depthwise separable convolution. It is implemented via the following steps: - Split the input into individual channels. - Convolve each channel with an individual depthwise kernel with `depth_multiplier` output channels. - Concatenate the convolved outputs along the channels axis. Unlike a regular convolution, depthwise convolution does not mix information across different input channels. The `depth_multiplier` argument determines how many filter are applied to one input channel. As such, it controls the amount of output channels that are generated per input channel in the depthwise step. Args: rank: int, the rank of the convolution, e.g. 2 for 2D convolution. depth_multiplier: The number of depthwise convolution output channels for each input channel. The total number of depthwise convolution output channels will be equal to `input_channel * depth_multiplier`. kernel_size: int or tuple/list of `rank` integers, specifying the size of the depthwise convolution window. strides: int or tuple/list of `rank` integers, specifying the stride length of the depthwise convolution. If only one int is specified, the same stride size will be used for all dimensions. `strides > 1` is incompatible with `dilation_rate > 1`. padding: string, either `"valid"` or `"same"` (case-insensitive). `"valid"` means no padding. `"same"` results in padding evenly to the left/right or up/down of the input. When `padding="same"` and `strides=1`, the output has the same size as the input. data_format: string, either `"channels_last"` or `"channels_first"`. The ordering of the dimensions in the inputs. `"channels_last"` corresponds to inputs with shape `(batch, steps, features)` while `"channels_first"` corresponds to inputs with shape `(batch, features, steps)`. It defaults to the `image_data_format` value found in your Keras config file at `~/.keras/keras.json`. If you never set it, then it will be `"channels_last"`. dilation_rate: int or tuple/list of `rank` integers, specifying the dilation rate to use for dilated convolution. If only one int is specified, the same dilation rate will be used for all dimensions. activation: Activation function. If `None`, no activation is applied. use_bias: bool, if `True`, bias will be added to the output. depthwise_initializer: Initializer for the depthwsie convolution kernel. If `None`, the default initializer (`"glorot_uniform"`) will be used. bias_initializer: Initializer for the bias vector. If `None`, the default initializer (`"zeros"`) will be used. depthwise_regularizer: Optional regularizer for the convolution kernel. bias_regularizer: Optional regularizer for the bias vector. activity_regularizer: Optional regularizer function for the output. depthwise_constraint: Optional projection function to be applied to the kernel after being updated by an `Optimizer` (e.g. used to implement norm constraints or value constraints for layer weights). The function must take as input the unprojected variable and must return the projected variable (which must have the same shape). Constraints are not safe to use when doing asynchronous distributed training. bias_constraint: Optional projection function to be applied to the bias after being updated by an `Optimizer`. c t|d||tj|d|||_||_t ||d|_t ||d|_t ||d|_ t||_ t||_ tj||_| |_t#j| |_t#j| |_tj| |_tj| |_t-j||_t-j||_t3|jdz|_|j|_ |j (|j dkrt7d|j d t9|jst7d |jd t9|jst7d |jt;|jd kDr>t;|jd kDr%t7d |jd|jyy)N) trainablenameactivity_regularizer kernel_sizestrides dilation_rate)min_ndimrzmInvalid value for argument `depth_multiplier`. Expected a strictly positive value. Received depth_multiplier=.zBThe argument `kernel_size` cannot contain 0. Received kernel_size=z;The argument `strides` cannot contains 0. Received strides=zW`strides > 1` not supported in conjunction with `dilation_rate > 1`. Received: strides=z and dilation_rate=)super__init__rgetrankdepth_multiplierr rrrr paddingr data_formatr activationuse_biasrdepthwise_initializerbias_initializerdepthwise_regularizerbias_regularizerrdepthwise_constraintbias_constraintr input_spec ValueErrorallmax)selfrr rrr!r"rr#r$r%r&r'r(rr)r*rrkwargs __class__s b/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/layers/convolutional/base_depthwise_conv.pyrzBaseDepthwiseConv.__init__TsD,  !-!1!12F!G    0,[$ N($ B . 4 +73 2;?%//*5  %1%5%56K%L" , 0 01A B%1%5%56K%L" , 0 01A B$/OO4H$I!*?#TYY];++  ,1F1F!1K$$($9$9#:!=  4##$#//03  4<< <<.*  t|| q S););%,,H!!%!3!3 46 &A c |jdk(rd}|d}nd}|d}t|jdz||i|_|j||j fz}|j d||j|j|jd|j|_ |jrT|j d |j |zf|j|j|jd|j|_yd|_y) N channels_lastrr)raxeskernelT)rshape initializer regularizer constraintrdtypebias)r"r rr+rr add_weightr%r'r)r=r8r$r&r(r*r>)r/ input_shape channel_axis input_channeldepthwise_shapes r2buildzBaseDepthwiseConv.builds    .L'OML'NM#YY], )F **   ! !.  oo!222200**&   ==,,}<> 11 11//jj(DIDIr3c<|jdk(r|d}|S|d}|S)Nr5r6r)r"r/r@rBs r2_get_input_channelz$BaseDepthwiseConv._get_input_channels2    .'OM(NMr3c<|j|j}tj||j|j |j |j|j}|jr|jdk(r$d|jdzz|j|zfz}n!d|j|zfd|jzz}tj|j|}tj||}|j|j|S|S)N)rr!rr"r5)rr)rGr9rdepthwise_convr8rr!rr"r$rr reshaper>addr#)r/inputsrBoutputs bias_shaper>s r2callzBaseDepthwiseConv.calls// = $$  KKLLLL,,((   ==?2!TYY]3))M97  !6!6!FGKIIK ;;tyy*5Dgggt,G ?? &??7+ +r3c |j|}t||j|z|j|j|j |j |jS)N)rr!r"r)rGr r rrr!r"rrFs r2compute_output_shapez&BaseDepthwiseConv.compute_output_shapesY// < (   ! !M 1   LLLL((,,  r3ct|}|j|j|j|j |j |j|jtj|j|jtj|jtj|jt!j|j"t!j|j$t!j|j&t)j|j*t)j|j,d|S)N)r rrr!r"rr#r$r%r&r'r(rr)r*)r get_configupdater rrr!r"rr serializer#r$rr%r&rr'r(rrr)r*)r/configr1s r2rSzBaseDepthwiseConv.get_configs#% $($9$9#//<<<<#//!%!3!3)33DOOD MM)5)?)?..*%1$:$:))%*6)?)?..*%1$:$:))%)5(>(>--))4(=(=--)$/#8#89M9M#N7  > r3)rvalidNrNTglorot_uniformzerosNNNNNTN) __name__ __module__ __qualname____doc__rrDrGrOrQrS __classcell__)r1s@r2rrs_AP. "!! 'HT"H6  !!r3rN)r] keras.srcrrrrrkeras.src.backendrkeras.src.layers.input_specr keras.src.layers.layerr keras.src.ops.operation_utilsr #keras.src.utils.argument_validationr r rrr3r2res78!!""51(CCAAAr3