tf.keras.layers.DepthwiseConv2D
| Просмотреть исходный код на GitHub |
Двумерная свертка по глубине.
tf.keras.layers.DepthwiseConv2D(
kernel_size,
strides=(1, 1),
padding='valid',
depth_multiplier=1,
data_format=None,
dilation_rate=(1, 1),
activation=None,
use_bias=True,
depthwise_initializer='glorot_uniform',
bias_initializer='zeros',
depthwise_regularizer=None,
bias_regularizer=None,
activity_regularizer=None,
depthwise_constraint=None,
bias_constraint=None,
**kwargs
)
Свертка по глубине — это тип свертки, в котором каждый входной канал свёртывается с разным ядром (называемым ядром по глубине). Вы можете рассматривать свертку по глубине как первый шаг в свёртке по разделению глубины.
Она реализуется следующим образом:
- Разделить вход на отдельные каналы.
- Свернуть каждый канал с отдельным ядром по глубине с
depth_multiplierвыходными каналами. - Конкатенировать свёрнутые выходы вдоль оси каналов.
В отличие от обычной двумерной свертки, свертка по глубине не смешивает информацию по разным входным каналам.
Аргумент depth_multiplier определяет количество фильтров, применяемых к одному входному каналу. Таким образом, он управляет количеством выходных каналов, генерируемых на каждый входной канал на этапе свертки по глубине.
| Аргументы | |
|---|---|
kernel_size | Целое число или кортеж/список из 2 целых чисел, определяющий высоту и ширину окна двумерной свертки. Может быть одним целым числом для задания одинакового значения для всех пространственных измерений. |
strides | Целое число или кортеж/список из 2 целых чисел, определяющий шаги свертки по высоте и ширине. Может быть одним целым числом для задания одинакового значения для всех пространственных измерений. Указание любого значения шага != 1 несовместимо с указанием любого значения dilation_rate != 1. |
padding | одно из 'valid' или 'same' (регистр не учитывается). "valid" означает отсутствие заполнения. "same" приводит к заполнению нулями равномерно слева/справа или сверху/снизу входных данных таким образом, что выход имеет те же размеры высоты/ширины, что и вход. |
depth_multiplier | Количество выходных каналов свертки по глубине для каждого входного канала. Общее количество выходных каналов свертки по глубине будет равно filters_in * depth_multiplier. |
data_format | Строка, одно из channels_last (по умолчанию) или channels_first. Порядок измерений на входах. channels_last соответствует входам с формой (batch_size, height, width, channels), а channels_first соответствует входам с формой (batch_size, channels, height, width). По умолчанию используется значение image_data_format в файле конфигурации Keras по адресу ~/.keras/keras.json. Если вы его никогда не устанавливали, он будет равен 'channels_last'. |
dilation_rate | Целое число или кортеж/список из 2 целых чисел, определяющий скорость расширения для свертки с расширением. В настоящее время указание любого значения dilation_rate != 1 несовместимо с указанием любого значения strides != 1. |
activation | Функция активации для использования. Если вы ничего не укажете, активация не применяется (см. keras.activations). |
use_bias | Булево значение, использует ли слой вектор смещения. |
depthwise_initializer | Инициализатор матрицы ядра свертки по глубине (см. keras.initializers). Если None, будет использоваться инициализатор по умолчанию ('glorot_uniform'). |
bias_initializer | Инициализатор вектора смещения (см. keras.initializers). Если None, будет использоваться инициализатор по умолчанию ('zeros'). |
depthwise_regularizer | Функция регуляризации, применяемая к матрице ядра свертки по глубине (см. keras.regularizers). |
bias_regularizer | Функция регуляризации, применяемая к вектору смещения (см. keras.regularizers). |
activity_regularizer | Функция регуляризации, применяемая к выходу слоя (его 'активации') (см. keras.regularizers). |
depthwise_constraint | Функция ограничения, применяемая к матрице ядра свертки по глубине (см. keras.constraints). |
bias_constraint | Функция ограничения, применяемая к вектору смещения (см. keras.constraints). |
Форма входа:
4D тензор с формой: [batch_size, channels, rows, cols] если data_format='channels_first' или 4D тензор с формой: [batch_size, rows, cols, channels] если data_format='channels_last'.
Форма выхода:
4D тензор с формой: [batch_size, channels * depth_multiplier, new_rows, new_cols] если data_format='channels_first' или 4D тензор с формой: [batch_size, new_rows, new_cols, channels * depth_multiplier] если data_format='channels_last'. Значения rows и cols могут измениться из-за заполнения.
| Возвращаемое значение | |
|---|---|
4-мерный тензор, представляющий activation(depthwiseconv2d(inputs, kernel) + bias). |
| Исключения | |
|---|---|
ValueError | если padding равно "causal". |
ValueError | при одновременном strides > 1 и dilation_rate > 1. |
Методы
convolution_op
convolution_op(
inputs, kernel
)
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/layers/DepthwiseConv2D