關於pytorch grid_sample()

torch.nn.functional.grid_sample(input, grid, mode=‘bilinear’, padding_mode=‘zeros’) 其中,grid的shape爲(N, H, W, 2) 對於output上的每一點,(x, y)三個通道的像素值,採集自input上某一點三個通道的像素值,採集的點存在於grid最低維,也就是(N, H, W, 2)中的2, [0]索
相關文章
相關標籤/搜索