PyTorch 数据加载报错：FileNotFoundError 解决指南

在使用 PyTorch 加载数据时，遇到 'FileNotFoundError' 是很常见的问题。这个错误通常表示代码无法找到指定的文件。本文将带你一步步解决这个问题，并提供一些实用的调试技巧。

错误分析

你提供的代码片段中，错误出现在 MyDataset 类的初始化过程中，具体来说是在尝试打开 self.names_file 文件时：pythonfile = open(self.names_file)

错误信息 'FileNotFoundError' 表明 Python 解释器无法在指定路径找到 self.names_file 文件。

解决步骤

1. 检查文件路径: 首先，你需要确认 args.root2 和 args.txtpath2 的值是否正确，确保这两个路径指向了文件实际存储的位置。

   • 打印 args.root2 和 args.txtpath2 的值，确认路径是否与预期相符。 - 检查路径中的拼写错误，尤其是目录分隔符 / 或 \ 是否使用正确。

2. 使用绝对路径: 建议使用绝对路径来代替相对路径，避免因为工作目录不同而导致文件无法找到。 你可以使用 os.path.abspath() 函数将相对路径转换为绝对路径。

3. 确认文件存在: 确保文件确实存在于指定的路径中。 你可以使用 os.path.exists() 函数检查文件是否存在。

4. 检查文件读取权限: 即使文件存在，你的代码也可能因为权限不足而无法读取文件。 确保你的代码拥有读取该文件的权限。

代码示例pythonimport osimport torchfrom torch.utils.data import DataLoaderfrom mydataset import MyDatasetimport argparse# ... 其他代码 ...

定义argsparser = argparse.ArgumentParser()parser.add_argument('--root2', type=str, default='path_to_root2', help='root directory of test dataset')parser.add_argument('--txtpath2', type=str, default='path_to_txtpath2', help='path to test dataset txt file')# ... 其他代码 ...args = parser.parse_args()

打印路径检查print('Root directory:', args.root2)print('Text file path:', args.txtpath2)

使用绝对路径root_dir = os.path.abspath(args.root2)txt_file_path = os.path.abspath(args.txtpath2)

检查文件是否存在if not os.path.exists(txt_file_path): raise FileNotFoundError(f'无法找到文件: {txt_file_path}')

创建数据集test_dataset = MyDataset(root_dir, txt_file_path, transform=None)# ... 其他代码 ...

通过以上步骤，你应该能够解决 'FileNotFoundError' 并成功加载数据。 记住，仔细检查代码、路径和文件权限是解决此类问题的关键。