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, необязательно) – если 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 (центр входного изображения).