如何在Python中实现PyTorch的线性插值_使用upsample或interpolate函数

落明同学_9041

落明同学_9041

2026-05-02

362人浏览

原创

优先用f.interpolate,它更灵活、参数统一且无额外可学习参数;nn.upsample仅适合需插入nn.sequential或复用的场景,二者底层逻辑相同但接口差异显著。

如何在python中实现pytorch的线性插值_使用upsample或interpolate函数

PyTorch里该用 F.interpolate 还是 nn.Upsample?

直接说结论:优先用 F.interpolate,它更灵活、参数统一,且不引入额外可学习参数;nn.Upsample 本质是封装了 F.interpolate 的模块,只适合需要插入到 nn.Sequential 或需复用的场景。

两者底层调用同一套插值逻辑,但接口差异明显:

  • F.interpolate 是函数式接口,输入张量、输出张量,所有插值控制(size、scale_factor、mode、align_corners)都显式传参
  • nn.Upsample 是类,初始化时固定 size 或 scale_factor 和 mode,调用时不能再改 —— 想换缩放倍数就得重新实例化
  • align_corners=False 是默认值,但对线性插值(mode="bilinear")影响显著:设为 True 时边界像素对齐,False 更符合多数视觉任务的坐标惯例(如 torchvision.transforms.Resize 默认 align_corners=False)

线性插值必须用 mode="bilinear" 吗?

不是。“线性插值”在 PyTorch 中对应的是多维推广:一维用 mode="linear",二维用 mode="bilinear",三维用 mode="trilinear"。选错 mode 会直接报错或结果异常。

常见错误现象:RuntimeError: expected 4D input (batch, channel, height, width) for bilinear mode —— 这说明你传入了 3D 张量却用了 bilinear。

  • 输入是 [N, C, H, W] → 用 mode="bilinear"
  • 输入是 [N, C, H](一维信号)→ 用 mode="linear",且必须确保 dim=2 是序列维度
  • 输入是 [N, C, D, H, W] → 用 mode="trilinear"
  • mode="nearest" 不是线性插值,别混用;"bicubic" 虽更平滑,但不是线性,计算开销也更大

怎么正确设置 size 和 scale_factor?

二者互斥:同时指定会报错 ValueError: only one of size or scale_factor should be defined。实际使用中,scale_factor 更常用,尤其做固定倍数上采样(如 ×2),而 size 更适合精确控制输出分辨率(如统一 resize 到 [224, 224])。

361SCAN security check your skill python scripts
361SCAN security check your skill python scripts

扫描并分析已安装技能,用于查看指定技能目录的名称、描述和详情,或列出所有已安装的技能。

下载
  • 用 scale_factor 时注意类型:整数或浮点数均可,但必须是标量(各维度同倍数)或长度为 2/3 的 tuple(如 scale_factor=(2.0, 1.5))
  • 用 size 时,必须匹配输入维度:对 [N,C,H,W],size 应为 (out_h, out_w) 或 torch.Size([out_h, out_w]),不能带 batch 和 channel
  • 当 scale_factor 导致非整数尺寸(如 H=100, scale_factor=1.5 → 150.0),PyTorch 会自动 floor 成 150,不报错但可能偏离预期

示例(上采样 2×):

import torch
import torch.nn.functional as F
<p>x = torch.randn(1, 3, 32, 32)
y = F.interpolate(x, scale_factor=2.0, mode='bilinear', align_corners=False)</p><h1>y.shape == torch.Size([1, 3, 64, 64])</h1><p></p>

为什么输出结果边缘模糊或偏移?

大概率是 align_corners 设置不当,或输入张量 layout 不符合插值假设。

  • align_corners=True:把输入的角点像素(如 (0,0) 和 (H-1,W-1))严格映射到输出的角点((0,0) 和 (H'-1,W'-1)),适合几何校准等需要像素级对齐的场景
  • align_corners=False(默认):把输入的角点映射到输出的边界中心(如 (0,0) → (0.5,0.5)),更符合卷积神经网络中特征图坐标的连续化建模习惯,也是大多数预训练模型(如 ResNet)期望的
  • 另一个常被忽略的点:PyTorch 插值默认按 NCHW 格式处理,如果你的数据是 NHWC(比如从某些 OpenCV 或 NumPy 流程来),必须先 permute(0,3,1,2),否则插值会在错误维度上发生

复杂点在于:同一个模型里不同层可能对 align_corners 敏感度不同;有些分割头要求 True 才能和 ground truth 对齐,而 backbone 特征上采样通常用 False。没有银弹,得看下游任务需求。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python pytorch

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1791

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4684

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1749

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

27777

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3347

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3347

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1203

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2603

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习