gpt4 book ai didi

pytorch - 在 Pytorch 中裁剪一小批图像——每个图像都不同

转载 作者:行者123 更新时间:2023-12-02 10:23:26 25 4
gpt4 key购买 nike

我有一个名为 input 的张量,尺寸为 64x21x21。它是 64 个图像的小批量,每个图像 21x21 像素。我想将每个图像裁剪为 11x11 像素。所以我想要的输出张量的尺寸为 64x11x11。

我想围绕不同的“中心像素”裁剪每个图像。中心像素由名为 center 的二维长张量给出,尺寸为 64x2。对于图像 i,center[i][0] 给出行索引,center[i][1] 给出应位于中心的像素的列索引在输出中。我们可以假设中心像素始终距离边界至少 5 个像素。

有没有一种有效的方法可以在 pytorch(在 GPU 上)中执行此操作?

更新:让我澄清一下,中心张量是由深度神经网络形成的。用强化学习术语来说,它充当“硬注意力机制”。 “裁剪”图像后,该子图像将成为另一个神经网络的输入。这就是为什么我想在Pytorch中进行裁剪:因为裁剪前后的操作都是在Pytorch中进行的。我希望避免将任何内容从 GPU 传输回 CPU。

最佳答案

我在 pytorch 论坛上提出了这个问题,并得到了答案。 grid_sample 函数应该可以完全解决问题。

https://discuss.pytorch.org/t/cropping-a-minibatch-of-images-each-image-a-bit-differently/12247

关于pytorch - 在 Pytorch 中裁剪一小批图像——每个图像都不同,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48235916/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com