Nvidia DIGITS - 超过前5名预测班级

Question

我正在使用nvidia-digits来训练CNN并获得预测。但是，.../classify_one.json默认返回前5个预测。我需要更大的n，如何修改我的请求参数以获取最高n？

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

Answer 1

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

这样的东西

accuracy_param {
  top_k: 5
}

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