如何使用PyTorch实现读取数据集的实战案例代码？

本文共计1087个文字，预计阅读时间需要5分钟。

目录概述项目结构与代码总结参考文献概述最近修复图像修正论文的代码配置环境后开始运行，发现数据直接加载不进去。嗯，看来还是得看人家家的代码写得好，一句一句地看逻辑。

目录

• 概述
• 项目结构与代码
• 总结
• 参考资料

概述

最近在跑一篇图像修复论文的代码，配置好环境之后开始运行，发现数据一直加载不进去。
害，还是得看人家代码咋写的，一句一句看逻辑，准能找出问题。通读dataset后，发现了问题所在，终于成功加载了数据集。

项目结构与代码

项目结构

主要的目的就是从数据集中读取到彩色图像和掩码图像。
代码
代码中涉及到torch.transforms、合并路径等知识点，我在代码中都进行了详细的注释，路径要对照着项目结构，如果自己用的话要根据项目结构去将相对路径改过来。