多类别适配问题

[gaowq2017]

作者您好，感谢您开源代码，有一些问题想向您请教一下：
代码所用数据集的Label的值为0（背景）和255（前景），而我自己的数据集的label值为0（背景）1、2、3......(多类别)，我修改了NCD和GRA，添加了类别数量参数，以及loss函数修改为CE，但是最终效果不理想，不知是否可以这么修改？或者这么修改存在什么问题？