F.interpolate插值大小设定

最新推荐文章于 2024-04-20 22:11:14 发布

撒旦即可

最新推荐文章于 2024-04-20 22:11:14 发布

阅读量1.5k

点赞数

分类专栏： Pytorch 文章标签：深度学习 python 人工智能

本文链接：https://blog.csdn.net/qq_39861441/article/details/125741090

版权

Pytorch 专栏收录该内容

36 篇文章 2 订阅

订阅专栏

import torch

import torch.nn.functional as F
h, w = 180, 320
img = torch.FloatTensor(1,3, h, w)
zoom=2
hr1 = F.interpolate(img, scale_factor=zoom, mode='bicubic')

hr2 = F.interpolate(img, (int(img.size()[2]*zoom),int(img.size()[3]*zoom)), mode='bicubic')

print(hr1.shape,hr2.shape)

输出 torch.Size([1, 3, 360, 640]) torch.Size([1, 3, 360, 640])