使用 NiftyNet 进行分类

Question

我正在尝试使用 NiftyNet 运行 class化。我想运行训练，我有一个训练集，其中包含图像和分配给它们的 class，例如图像 1 -> 癌症和图像 2 -> 无癌症。基本上我没有面具，看来如果我想运行 class 化，我需要有带有标记体素的面具才能运行训练 class 化。我假设这是基于以下页面：https://niftynet.readthedocs.io/en/dev/_modules/niftynet/application/classification_application.html#ClassificationApplication，它试图解释如何创建一个 classification 部分，该部分声明类似于分段。

最终，我的问题是，如果我只有图像和 class 每个图像所属的图像，我是否可以运行 class 与 NiftyNet 化。例如。如果我有 4 classes 0、1、2 和 3，并且我已经映射了我的数据，例如 image1 -> 1，image2 -> 3，image3 -> 1，image4 -> 2，image5 -> 0 等。我可以将图像用作输入并输出向量吗？ [0 0 1 0] 用 NiftyNet 训练？如果是这样，您能否举例说明如何在 .ini 文件中创建 CLASSIFICATION 部分？

提前抱歉这个问题太长了。

Answer 1

事实上，现在您应该创建一个大小为 (1,1,1) 的 nifti 文件，其中包含您尝试预测的标签。
我认为你所说的向量是一个单一的热编码器，一旦标签被加载，就会被 niftynet 处理。这是分类的输入部分

[CLASSIFICATION]
image = Rx
label = labels
output_prob = True
num_classes = 2
label_normalisation = False

此致

使用 NiftyNet 进行分类

Classification with NiftyNet

niftynet