image-segmentation - NiftyNet 找不到用于多类分割的多个标签

Question

我正在运行一个 3 类分割问题，我的标签图像是“0（很多），1（一些），2（一些）”。但我意识到当 NiftyNet 读取图像时，它只有 0/1 两个值。当 label_normalization 开启时，ref.txt 文件显示“label from 0,1 to 0,1”；如果我把它关掉，预测也只有 0/1 两个值。

知道出了什么问题吗？标签图像格式问题？

非常感谢！

Answer 1

我可以问一下num_classes您正在使用的配置文件中的内容吗？当 NiftyNet 生成 时label_normalization，它会遍历它找到的所有文件。您如何确定文件列表？如果是通过搜索路径，您确定路径上有所有相关文件吗？