как дать padding = 6 в tensorflow 2.0?
self.conv6 = tf.keras.layers.Conv2D(1024, kernel_size=3, padding=6,dilation_rate=6) # atrous convolution
это дает ошибку в tensorflow 2.0.
Я вижу параметр padding='same',
но что, если я хочу указать padding=6
Файл "/home/jake/venv/lib/python3.7/site-packages/tensorflow_core/python/keras/utils/conv_utils.py", строка 202, в normalize_padding padding = value.lower() AttributeError: 'int' object не имеет атрибута "ниже"
1 ответ
Согласно tf.keras.layers.Conv2D
документация, padding
может быть только 'same'
или 'valid'
.
'same'
обеспечивает сохранение пространственных размеров и 'valid'
добавляет минимальное количество отступов, необходимых для обеспечения того, чтобы фильтр "соответствовал" вашим пространственным размерам (что может быть проблемой, если stride
есть что-нибудь кроме одного).
Вам не нужно указывать нестандартный размер отступа, поскольку в большинстве случаев повышающая дискретизация будет ненужной. Но при необходимости вы можете иметь слой заполнения перед слоем свертки.
tf.keras.layers.ZeroPadding2D(padding=(6, 6)) # put this before your conv layer