积分商城

最新搜索： 74HC165应用电路图 单相SVPWM AVR和PROTEUS实训100例2 Qtouch gd32f103 uc RTD2513
您现在的位置是：首页 > 技术资料 > PyTorch常用工具模块
推荐星级：
  • 1
  • 2
  • 3
  • 4
  • 5

PyTorch常用工具模块

更新时间：2026-07-01 08:17:51 大小：18K 上传用户：潇潇江南查看TA发布的资源 标签：pytorch 下载积分：2分 评价赚积分 （如何评价?） 打赏 收藏 评论(0) 举报

资料介绍

torchvision.io模块：数据输入输出工具

torchvision.io模块主要负责各类数据的读取与写入操作，支持图像、视频等多媒体格式的处理，是PyTorch数据 pipeline 的起始环节，能直接将外部文件转换为PyTorch张量格式，省去额外的格式转换步骤。

1. 图像读写功能

torchvision.io中的read_image和write_jpeg/write_png是最常用的图像读写接口，原生支持PyTorch张量输出，无需经过PIL转numpy再转张量的多余步骤。

核心函数说明：

· torchvision.io.read_image(path: str, mode: ImageReadMode = ImageReadMode.UNCHANGED) -> Tensor

参数说明：

o path：图像文件的本地路径，支持JPEG、PNG、BMP等常见格式

o mode：读取模式，共有三种可选：

1. ImageReadMode.UNCHANGED：保持原图通道数不变，原图为灰度则输出单通道，为RGBA则输出4通道

2. ImageReadMode.GRAY：强制转换为单通道灰度图像

3. ImageReadMode.RGB：强制转换为3通道RGB图像，是绝大多数视觉任务的默认读取模式

返回值：形状为(C, H, W)的torch.uint8类型张量，完全符合PyTorch图像张量的维度约定，可直接送入模型预处理流程。

· torchvision.io.write_jpeg(input: Tensor, path: str, quality: int = 75) -> None

用于将3维度RGB张量保存为JPEG格式图像，quality参数控制压缩质量，取值范围1-100，数值越大质量越高、文件体积越大。

· torchvision.io.write_png(input: Tensor, path: str, compression_level: int = 6) -> None

用于将单通道灰度或3通道RGB张量保存为PNG格式，支持无损压缩，compression_level取值范围0-9，数值越大压缩率越高、保存速度越慢。


部分文件列表

文件名 大小
PyTorch常用工具模块.docx 18K

相关下载

全部评论(0)

暂无评论
评论赚积分>>

上传资源 上传优质资源有赏金

热门标签

更多>>

最新上传

  • 打赏
  • 30日榜单

热门下载

资料专题

推荐下载

本站上的所有资源均为源于网上收集或者由用户自行上传，仅供学习和研究使用，无任何商业目的，版权归原作如有侵权，请 来信指出，本站将立即改正。

ICP许可证号：京ICP证070360号     21IC电子网 2000- 版权所有

京ICP备11013301号

京公网安备 11010802024343号