Nvidia DIGITS - 超过预测的前 5 名 类

Nvidia DIGITS - more than top 5 predicted classes

我正在使用 训练 CNN 并获得预测。但是默认情况下 .../classify_one.json returns 前 5 个预测。我需要一个更大的 n,如何修改我的请求参数以获得顶部 n 例如?

我在 DIGITS 文档中找不到任何内容。

尝试修改模型 prototxt 文件的顶 "Accuracy" 层的 accuracy_param。你应该有类似

的东西
accuracy_param {
  top_k: 5
}

top_k: 5 更改为您希望获得的数字,例如 top_k: 10 以获得前 10 名 类。