- 1
- 2
- 3
- 4
- 5
PyTorch常用工具模块
资料介绍
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 |
最新上传
- 打赏
- 30日榜单
-
emlimei 打赏1.00元 1天前
-
21ic小能手 打赏5.00元 1天前
-
21ic小能手 打赏3.00元 2天前
资料：低频功率放大器的设计.
-
21ic小能手 打赏3.00元 2天前
-
21ic小能手 打赏3.00元 2天前
-
21ic小能手 打赏3.00元 2天前
-
21ic小能手 打赏3.00元 2天前
-
21ic小能手 打赏5.00元 3天前
资料：51单片机数字频率计
-
21ic小能手 打赏5.00元 3天前
-
21ic小能手 打赏5.00元 3天前
-
21ic小能手 打赏5.00元 3天前
-
13573902396 打赏1.00元 3天前
-
21下载积分 打赏310.00元 3天前
用户：江岚
-
21下载积分 打赏310.00元 3天前
用户：潇潇江南
-
21下载积分 打赏60.00元 3天前
用户：他山之石可攻玉
-
21下载积分 打赏310.00元 3天前
用户：小猫做电路
-
21下载积分 打赏210.00元 3天前
用户：zhengdai
-
21下载积分 打赏210.00元 3天前
用户：w993263495
-
21下载积分 打赏10.00元 3天前
用户：烟雨
-
21下载积分 打赏60.00元 3天前
用户：gsy幸运
-
21下载积分 打赏70.00元 3天前
用户：铁蛋锅
-
21下载积分 打赏65.00元 3天前
用户：xzxbybd
-
21下载积分 打赏60.00元 3天前
用户：jh0355
-
21下载积分 打赏60.00元 3天前
用户：w178191520
-
21下载积分 打赏20.00元 3天前
用户：jh03551
-
21下载积分 打赏20.00元 3天前
用户：sun2152
-
21下载积分 打赏20.00元 3天前
用户：kk1957135547
-
21下载积分 打赏25.00元 3天前
用户：w1966891335
-
21下载积分 打赏20.00元 3天前
用户：xuzhen1
-
21下载积分 打赏15.00元 3天前
用户：x15580286248
-
21下载积分 打赏25.00元 3天前
用户：pcb
-
21下载积分 打赏20.00元 3天前
用户：bhacker
-
21下载积分 打赏15.00元 3天前
用户：liqiang9090
-
21下载积分 打赏25.00元 3天前
用户：有理想666
-
21下载积分 打赏15.00元 3天前
用户：godbox
-
21下载积分 打赏15.00元 3天前
用户：aetek
-
21下载积分 打赏5.00元 3天前
用户：mulanhk
-
21下载积分 打赏5.00元 3天前
用户：JuneLin61
-
21下载积分 打赏5.00元 3天前
用户：ccc6188
-
21下载积分 打赏5.00元 3天前
用户：木集盒
全部评论(0)