时间戳

首页 > 解决方案 > 如何使用pytorch对图像进行双三次上采样？

python - 如何使用pytorch对图像进行双三次上采样？

问题描述

我有png图像。我想使用双三次插值对其进行上采样。我在 pytorch 中找到了这个函数：

nn.functional.upsample(mode = "bicubic)

https://pytorch.org/docs/stable/generated/torch.nn.Upsample.html

但是如何应用于我的 png 图像？我应该把我的形象变成一些火炬张量吗？我只是没有找到任何在 png 图像上完全使用此功能的示例

标签： pythonpython-3.ximagepytorch

解决方案

你可以这样做

import torch 
import torchvision.transforms as transforms
from PIL import Image


t = transforms.ToTensor()
img = Image.open("Table.png")

b = torch.nn.functional.upsample(t(img).unsqueeze(0),(500,400),mode = "bicubic")

您还可以使用图像应用双三次

img = Image.open("Table.png")
re = img.resize((400, 400),Image.BICUBIC)

推荐阅读