如何准备自定义数据集?

我使用掩膜图像作为标签(背景为 0，前景为 255)，但当 "--label_nc" 为 2 或 256 时，它无法正常工作。只有当我更改掩膜图像(背景为 0，前景为 1)，并设置 "--label_nc" 为 2 时，它才能正常工作。我不知道为什么它能正常工作。是否是自定义数据集的问题？