我正在尝试创建用于在 2D 世界中训练机器人的数据集。 世界将有诸如“自由”、“未知”、“障碍”之类的标签来识别地图的不同组成部分。 这很可能是使用整数作为标签的矩阵，例如免费=1，未知=2 等
我想用这个数据集来训练一个 CNN 来学习地图中每个状态的值。 当前的计划是将这个世界分解为多个二元矩阵以分离不同的标签。 所以，最终神经网络会有像输入 RGB 图像这样的输入（但这里不同的矩阵层将代表不同的标签）。
我查看了 Pytorch 中的 DataLoader 类，它允许我们创建自定义数据集。 但是，在我的数据集中，我没有 .jpg 文件，而是有单独的矩阵层来表示地图中每个州的不同标签。 有谁知道如何在 PyTorch 中创建我想要的数据集？