o 7?e:@sdZddlmmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZed d GdddeZeZdS)zGKeras 3D transposed convolution layer (sometimes called deconvolution).N) activations) constraints) initializers) regularizers)utils) InputSpec)Conv3D) conv_utils) keras_exportzkeras.layers.Conv3DTransposez#keras.layers.Convolution3DTransposecsheZdZdZej              dfdd Zd d Zd d ZddZ fddZ Z S)Conv3DTransposeaTransposed convolution layer (sometimes called Deconvolution). The need for transposed convolutions generally arises from the desire to use a transformation going in the opposite direction of a normal convolution, i.e., from something that has the shape of the output of some convolution to something that has the shape of its input while maintaining a connectivity pattern that is compatible with said convolution. When using this layer as the first layer in a model, provide the keyword argument `input_shape` (tuple of integers or `None`, does not include the sample axis), e.g. `input_shape=(128, 128, 128, 3)` for a 128x128x128 volume with 3 channels if `data_format="channels_last"`. Args: filters: Integer, the dimensionality of the output space (i.e. the number of output filters in the convolution). kernel_size: An integer or tuple/list of 3 integers, specifying the depth, height and width of the 3D convolution window. Can be a single integer to specify the same value for all spatial dimensions. strides: An integer or tuple/list of 3 integers, specifying the strides of the convolution along the depth, height and width. Can be a single integer to specify the same value for all spatial dimensions. Specifying any stride value != 1 is incompatible with specifying any `dilation_rate` value != 1. padding: one of `"valid"` or `"same"` (case-insensitive). `"valid"` means no padding. `"same"` results in padding with zeros evenly to the left/right or up/down of the input such that output has the same height/width dimension as the input. output_padding: An integer or tuple/list of 3 integers, specifying the amount of padding along the depth, height, and width. Can be a single integer to specify the same value for all spatial dimensions. The amount of output padding along a given dimension must be lower than the stride along that same dimension. If set to `None` (default), the output shape is inferred. data_format: A string, one of `channels_last` (default) or `channels_first`. The ordering of the dimensions in the inputs. `channels_last` corresponds to inputs with shape `(batch_size, depth, height, width, channels)` while `channels_first` corresponds to inputs with shape `(batch_size, channels, depth, height, width)`. When unspecified, uses `image_data_format` value found in your Keras config file at `~/.keras/keras.json` (if exists) else 'channels_last'. Defaults to 'channels_last'. dilation_rate: an integer or tuple/list of 3 integers, specifying the dilation rate to use for dilated convolution. Can be a single integer to specify the same value for all spatial dimensions. Currently, specifying any `dilation_rate` value != 1 is incompatible with specifying any stride value != 1. activation: Activation function to use. If you don't specify anything, no activation is applied (see `keras.activations`). use_bias: Boolean, whether the layer uses a bias vector. kernel_initializer: Initializer for the `kernel` weights matrix (see `keras.initializers`). Defaults to 'glorot_uniform'. bias_initializer: Initializer for the bias vector (see `keras.initializers`). Defaults to 'zeros'. kernel_regularizer: Regularizer function applied to the `kernel` weights matrix (see `keras.regularizers`). bias_regularizer: Regularizer function applied to the bias vector (see `keras.regularizers`). activity_regularizer: Regularizer function applied to the output of the layer (its "activation") (see `keras.regularizers`). kernel_constraint: Constraint function applied to the kernel matrix (see `keras.constraints`). bias_constraint: Constraint function applied to the bias vector (see `keras.constraints`). Input shape: 5D tensor with shape: `(batch_size, channels, depth, rows, cols)` if data_format='channels_first' or 5D tensor with shape: `(batch_size, depth, rows, cols, channels)` if data_format='channels_last'. Output shape: 5D tensor with shape: `(batch_size, filters, new_depth, new_rows, new_cols)` if data_format='channels_first' or 5D tensor with shape: `(batch_size, new_depth, new_rows, new_cols, filters)` if data_format='channels_last'. `depth` and `rows` and `cols` values might have changed due to padding. If `output_padding` is specified:: ``` new_depth = ((depth - 1) * strides[0] + kernel_size[0] - 2 * padding[0] + output_padding[0]) new_rows = ((rows - 1) * strides[1] + kernel_size[1] - 2 * padding[1] + output_padding[1]) new_cols = ((cols - 1) * strides[2] + kernel_size[2] - 2 * padding[2] + output_padding[2]) ``` Returns: A tensor of rank 5 representing `activation(conv3dtranspose(inputs, kernel) + bias)`. Raises: ValueError: if `padding` is "causal". ValueError: when both `strides` > 1 and `dilation_rate` > 1. References: - [A guide to convolution arithmetic for deep learning](https://arxiv.org/abs/1603.07285v1) - [Deconvolutional Networks](https://www.matthewzeiler.com/mattzeiler/deconvolutionalnetworks.pdf) r r validNTglorot_uniformzerosc stjd ||||||t|| t| t| t| t| t|t|t|d|||_|jduratj |jdddd|_t |j |jD]\}}||kr`t d|j d|jdqKdSdS) N)filters kernel_sizestridespadding data_format dilation_rate activationuse_biaskernel_initializerbias_initializerkernel_regularizerbias_regularizeractivity_regularizerkernel_constraintbias_constraintoutput_paddingT)Z allow_zeroz>Strides must be greater than output padding. Received strides=z, output_padding=.) super__init__rgetrrrr!r Znormalize_tuplezipr ValueError)selfrrrrr!rrrrrrrrrrrkwargsstrideZout_pad __class__r#p/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/keras/src/layers/convolutional/conv3d_transpose.pyr%sH  zConv3DTranspose.__init__c Cst|}t|dkrtd|d|}|j|jdur*td|d|dt||}|j|j |f}t d||id|_ |j d||j |j|jd |jd |_|jri|j d |j f|j|j|jd |jd |_nd|_d |_dS) Nz0Inputs should have rank 5. Received input_shape=r"zhThe channel dimension of the inputs to `Conv3DTranspose` should be defined. The input_shape received is z , where axis z> (0-based) is the channel dimension, which found to be `None`.)ndimZaxeskernelT)shapeZ initializerZ regularizer constraintZ trainabledtypebias)tf TensorShapelenr(Z_get_channel_axisdimsvalueintrrrZ input_specZ add_weightrrrr4r1rrrrr5Zbuilt)r) input_shapeZ channel_axisZ input_dimZ kernel_shaper#r#r.buildsP      zConv3DTranspose.buildc Cst|}|d}|jdkrd\}}}nd\}}}||}||}||} |j\} } } |j\} }}|jdur=d}}}n|j\}}}tj|| |j|| d}tj|| |j||d}tj| | |j||d}|jdkry||j |||f}dd| ||f}n|||||j f}d| ||df}t |}tj j ||j ||tj|jdd|jd }ts|jjr||j}|||jrtj j||jtj|jd dd }|jdur||S|S) Nrchannels_first)r )r r?r rr!r+r r/)r0)rrr@)r)r6r2rrrr!r deconv_output_lengthrrstacknnZconv3d_transposer1Zconvert_data_formatupperZexecuting_eagerlyZrankcompute_output_shape set_shaperZbias_addr5r)r)ZinputsZ inputs_shapeZ batch_sized_axish_axisw_axisdepthheightwidthkernel_dkernel_hkernel_wstride_dstride_hstride_w out_pad_d out_pad_h out_pad_wZ out_depthZ out_heightZ out_width output_shaperZoutput_shape_tensorZoutputsZ out_shaper#r#r.calls              zConv3DTranspose.callcCst|}t|}|jdkrd\}}}}nd\}}}}|j\}}} |j\} } } |jdur5d} }}n|j\} }}|j||<t j ||||j | | d||<t j ||||j || d||<t j ||| |j || d||<t|S)Nr>)r r?r r@)r@r r?r rA) r6r7as_listlistrrrr!rr rBr)r)r<rWZc_axisrHrIrJrNrOrPrQrRrSrTrUrVr#r#r.rFWsB          z$Conv3DTranspose.compute_output_shapecs"t}|d|j|d<|S)Nrr!)r$ get_configpopr!)r)configr,r#r.r[s   zConv3DTranspose.get_config)r rNNr NTrrNNNNN) __name__ __module__ __qualname____doc__rZallow_initializer_layoutr%r=rXrFr[ __classcell__r#r#r,r.r !s,w4+[(r )raZtensorflow.compat.v2compatv2r6Z keras.srcrrrrZkeras.src.dtensorrZkeras.src.engine.input_specrZ%keras.src.layers.convolutional.conv3drZkeras.src.utilsr Z tensorflow.python.util.tf_exportr r ZConvolution3DTransposer#r#r#r.s$         f