用NiftyNet进行分类

Question

我正在尝试使用NiftyNet运行分类。我想要跑步训练，我有一个训练集，图像和一个课程分配给他们，如image1 ->癌症和image2 -> no癌。基本上，我没有口罩，如果我想要运行分类，我需要有标记体素的面具来进行分类训练。我假设这是基于以下页面：application.html#ClassificationApplication，它试图解释如何创建一个类似于分段的分类部分。

最终，我的问题是，如果只有图像和每个图像所属的类，那么我是否能够使用NiftyNet运行分类。例如，如果我有4个类0、1、2和3，并且映射了我的数据，例如image1 -> 1、image2 -> 3、image3 -> 1、image4 -> 2、image5 -> 0等。我是否可以使用图像作为输入，并输出向量(如0 0 1 0)，用于NiftyNet的训练？如果是这样的话，您能给出一个如何在.ini文件中创建分类部分的示例吗？

很抱歉这个问题太长了。

Answer 1

实际上，现在您应该创建一个大小为(1,1,1)的nifti文件，其中包含您试图在其中预测的标签。

我认为你所说的向量是一个热编码器，一旦标签被加载，niftynet就会处理它。这是分类的输入部分

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

诚挚的问候