您好，关于多类别数据集的训练

[wangqingl]

你好，作者，感谢您的工作内容。我想问一下，我发现了一个数据集，是将变化进行了分类，如“从沙漠变化到河流→1”“从沙漠变化到建筑→2”，这样的话您的代码如何修改呢？

[Bobholamovic]

你好，如果你指的是P2V模型的话，这个模型目前暂时不支持多类变化检测。不过，你可以通过修改这里的代码： https://github.com/Bobholamovic/CDLab/blob/bb6dc5eee2b1b259cebdcd0cf67de796a111a4ca/src/models/p2v.py#L188

调整输出通道数。此外，需要编写新的数据集接口，损失函数也需要针对多类变化检测做相应调整。