当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.keras.layers.Conv3D用法及代码示例


3D 卷积层(例如,体积上的空间卷积)。

继承自:LayerModule

用法

tf.keras.layers.Conv3D(
    filters, kernel_size, strides=(1, 1, 1), padding='valid',
    data_format=None, dilation_rate=(1, 1, 1), groups=1, activation=None,
    use_bias=True, kernel_initializer='glorot_uniform',
    bias_initializer='zeros', kernel_regularizer=None,
    bias_regularizer=None, activity_regularizer=None, kernel_constraint=None,
    bias_constraint=None, **kwargs
)

参数

  • filters 整数,输出空间的维度(即卷积中输出滤波器的数量)。
  • kernel_size 一个整数或 3 个整数的元组/列表,指定 3D 卷积窗口的深度、高度和宽度。可以是单个整数,为所有空间维度指定相同的值。
  • strides 一个整数或 3 个整数的元组/列表,指定卷积沿每个空间维度的步长。可以是单个整数,为所有空间维度指定相同的值。指定任何步幅值 != 1 与指定任何 dilation_rate 值 != 1 不兼容。
  • padding "valid""same" 之一(不区分大小写)。 "valid" 表示没有填充。 "same" 导致在输入的左/右或上/下均匀填充零,以使输出具有与输入相同的高度/宽度尺寸。
  • data_format 一个字符串,是 channels_last (默认)或 channels_first 之一。输入中维度的排序。 channels_last 对应于形状为 batch_shape + (spatial_dim1, spatial_dim2, spatial_dim3, channels) 的输入,而 channels_first 对应于形状为 batch_shape + (channels, spatial_dim1, spatial_dim2, spatial_dim3) 的输入。它默认为您的 Keras 配置文件中的 image_data_format~/.keras/keras.json 。如果您从未设置它,那么它将是"channels_last"。
  • dilation_rate 一个整数或 3 个整数的元组/列表,指定用于扩张卷积的扩张率。可以是单个整数,为所有空间维度指定相同的值。目前,指定任何 dilation_rate 值 != 1 与指定任何步幅值 != 1 不兼容。
  • groups 一个正整数,指定输入沿通道轴拆分的组数。每个组分别与 filters / groups 过滤器进行卷积。输出是沿通道轴的所有groups 结果的串联。输入通道和 filters 都必须能被 groups 整除。
  • activation 要使用的激活函数。如果您未指定任何内容,则不会应用激活(请参阅keras.activations)。
  • use_bias 布尔值,层是否使用偏置向量。
  • kernel_initializer kernel 权重矩阵的初始化程序(参见 keras.initializers )。默认为'glorot_uniform'。
  • bias_initializer 偏置向量的初始化程序(参见keras.initializers)。默认为'zeros'。
  • kernel_regularizer 应用于kernel 权重矩阵的正则化函数(参见keras.regularizers)。
  • bias_regularizer 应用于偏置向量的正则化函数(参见keras.regularizers)。
  • activity_regularizer 应用于层输出的正则化函数(其"activation")(参见keras.regularizers)。
  • kernel_constraint 应用于核矩阵的约束函数(参见keras.constraints)。
  • bias_constraint 应用于偏置向量的约束函数(参见keras.constraints)。

返回

  • 代表 activation(conv3d(inputs, kernel) + bias) 的 5+ 阶张量。

抛出

  • ValueError 如果padding 是"causal"。
  • ValueError strides > 1dilation_rate > 1 时。

该层创建一个卷积核,该卷积核与层输入进行卷积以产生输出张量。如果 use_bias 为 True,则会创建一个偏置向量并将其添加到输出中。最后,如果 activation 不是 None ,它也会应用于输出。

当将此层用作模型中的第一层时,请提供关键字参数 input_shape(整数元组或 None ,不包括样本轴),例如input_shape=(128, 128, 128, 1) 用于具有单通道的 128x128x128 卷,在 data_format="channels_last" 中。

例子:

# The inputs are 28x28x28 volumes with a single channel, and the
# batch size is 4
input_shape =(4, 28, 28, 28, 1)
x = tf.random.normal(input_shape)
y = tf.keras.layers.Conv3D(
2, 3, activation='relu', input_shape=input_shape[1:])(x)
print(y.shape)
(4, 26, 26, 26, 2)
# With extended batch shape [4, 7], e.g. a batch of 4 videos of 3D frames,
# with 7 frames per video.
input_shape = (4, 7, 28, 28, 28, 1)
x = tf.random.normal(input_shape)
y = tf.keras.layers.Conv3D(
2, 3, activation='relu', input_shape=input_shape[2:])(x)
print(y.shape)
(4, 7, 26, 26, 26, 2)

输入形状:

具有形状的 5+D 张量:batch_shape + (channels, conv_dim1, conv_dim2, conv_dim3)如果 data_format='channels_first' 或具有形状的 5+D 张量:batch_shape + (conv_dim1, conv_dim2, conv_dim3, channels)如果data_format='channels_last'。

输出形状:

5+D 张量,形状:batch_shape + (filters, new_conv_dim1, new_conv_dim2, new_conv_dim3) if data_format='channels_first' 或 5+D 张量,形状:batch_shape + (new_conv_dim1, new_conv_dim2, new_conv_dim3, filters) if data_format='channels_last'。 new_conv_dim1new_conv_dim2new_conv_dim3 值可能由于填充而发生了变化。

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.layers.Conv3D。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。