关于grid_sample这个函数,我没有在pytorch的中文文档中找到,在英文文档里面是有的:https://pytorch.org/docs/stable/nn.html#torch-nn-functional。
grid_sample这个函数的作用,是将一个source_image,通过双线性插值的方式变换到另一个大小指定的target_image中。
torch.nn.functional.grid_sample(input, grid, mode=‘bilinear’, padding_mode=‘zeros’)
第一个参数input为输入的图片。
第二个参数grid(Batch,H,W,2)为要将输入图片变换为的图片大小,最后的2并不是很明白是什么意思。
第三、第四个参数一般为默认值。
grid_sample其实就是实现了一个双线性插值的图片大小变换。
torch.nn.functional.grid_sample
最新推荐文章于 2025-03-26 14:30:44 发布