torch.nn.functional.affine_grid

torch.nn.functional.affine_grid(theta, size, align_corners=None) [source]

Создаёт 2D или 3D поле потока (выборочную сетку) на основе набора аффинных матриц theta.

Примечание

Эта функция часто используется в сочетании с grid_sample() для построения сетей пространственных преобразований .

Параметры:

• theta (Тензор) – входной пакет аффинных матриц с формой ($N \times 2 \times 3$) для 2D или ($N \times 3 \times 4$) для 3D
• size (torch.Size) – целевой размер изображения вывода. ($N \times C \times H \times W$ для 2D или $N \times C \times D \times H \times W$ для 3D) Пример: torch.Size((32, 3, 24, 24))
• align_corners (bool, optional) – если True, учитывать -1 и 1 для указания на центры угловых пикселей, а не углов изображения. Более подробное описание см. в grid_sample(). Сгенерированная с помощью affine_grid() сетка должна передаваться в grid_sample() с тем же значением этого параметра. По умолчанию: False

Возвращаемые значения:

Выходной тензор размера ($N \times H \times W \times 2$)

Тип возвращаемого значения:

выход (Тензор)

Предупреждение

При align_corners = True, позиции сетки зависят от размера пикселя относительно размера входного изображения, и поэтому местоположения, выбираемые с помощью grid_sample(), будут отличаться для одного и того же входного изображения при разных разрешениях (то есть после увеличения или уменьшения размера). Поведение по умолчанию до версии 1.2.0 было align_corners = True. С тех пор поведение по умолчанию было изменено на align_corners = False, чтобы соответствовать значению по умолчанию для interpolate().

Предупреждение

Когда align_corners = True, 2D аффинные преобразования 1D данных и 3D аффинные преобразования 2D данных (то есть, когда один из пространственных размеров имеет единичный размер) не определены и не предназначены для использования. Это не проблема, когда align_corners = False. До версии 1.2.0 все точки сетки вдоль единичного измерения условно рассматривались как -1. Начиная с версии 1.3.0, в условиях align_corners = True все точки сетки вдоль единичного измерения рассматриваются как 0 (центр входного изображения).