给定一批图像,我想提取所有可能的图像 block ,类似于卷积。在 TensorFlow 中,我们可以使用 tf.extract_image_patches
为达到这个。 PyTorch 中是否有等效的函数?
谢谢。
最佳答案
不幸的是,可能没有直接的方法来实现您的目标。
但 Tensor.unfold 函数可能是一个解决方案。
https://discuss.pytorch.org/t/how-to-extract-smaller-image-patches-3d/16837/2
该网站可能对您有所帮助。
关于python - PyTorch 中是否有提取图像补丁的功能?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45828265/