python - 使用自定义数据集在 PyTorch 中加载数据

Question

我正在尝试使用 PyTorch 中的 KAIST 数据集来训练模型并生成权重。我是 python 和 PyTorch 的新手，我正在努力使用带有这样一个数据集的数据加载器。该数据集带有 png 图像文件，但注释（标签和边界框）采用 .vbb 格式，我无法将其转换为可用形式（即 csv、文本文件等）。所以我想我的第一个问题是，我如何将注释转换为我可以使用它们的关联注释迭代 png 文件的格式。第二个问题是，我将如何在 PyTorch 中为这个数据集创建数据加载器。

任何建议将不胜感激。

链接到数据集 - https://github.com/SoonminHwang/rgbt-ped-detection/tree/master/data

有一种方法可以转换上面链接中提供的注释，但我无法让它产生正确的结果。也许有一种更简单的通过python进行转换的方法？

标签： pythonpytorch