PyTorch grid_sample返回零数组


Innate Mind
2025-03-18 09:17:44 (3天前)


我想通过affine_grid / grid_sample从我的图像中采样矩形贴片
我创建了只包含255个值的数组

canvas1 = np.zeros((128,128),dtype = np.uint8)
canvas1 [:] = 255
我也……

2 条回复
  1. 0# 尼拉。 | 2019-08-31 10-32



    您的网格值在[-1,1]之外。



    根据

    https://pytorch.org/docs/stable/nn.html#torch.nn.functional.grid_sample

    ,这些值按照定义处理

    padding_mode




    默认

    padding_mode

    是’零’,你可能想要的是“边界”:

    F.grid_sample(canvas1_torch, grid, mode=”bilinear”, padding_mode=”border”)

    返回所有值255。


登录 后才能参与评论