语义切分中的输出

Question

我在做Kaggle语义分割任务

在我代码的测试部分

model = model.eval()
predictions =[]
for data in testdataloader:
    data = t.autograd.Variable(data, volatile=True).cuda()
    output = model.forward(data)
    _,preds = t.max(output, 1, keepdim = True)

当我做preds部分时，数组中只填充了，我希望它是一个最大位置的数组，我不知道出了什么问题。 output部分工作得很好，我附加了一个用于可视化output的屏幕截图

任何关于出了什么问题的建议都会很有帮助。你知道吗

谢谢

Answer 1

假设您的数据是MiniBatch x Dim形式的，那么您现在要做的就是查看哪个minibatch的值最高。如果您使用单个样本（MB = 1）进行测试，那么您将始终得到0作为您的答案。因此，您可能需要尝试：

_,preds = t.max(output, 0, keepdim = False)