leon 5 月之前
父节点
当前提交
fee6dc9a09
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/resnet.cu

+ 1 - 1
src/resnet.cu

@@ -417,7 +417,7 @@ class InferImpl : public Infer {
       arrout.emplace_back(max_score, index);
       arrout.emplace_back(max_score, index);
     }
     }
     for (int ib = 0; ib < num_image; ++ib) {
     for (int ib = 0; ib < num_image; ++ib) {
-        std::cout << "score : " << arrout[i].confidence << " label : " << arrout[i].class_label << std::endl;
+        std::cout << "score : " << arrout[ib].confidence << " label : " << arrout[ib].class_label << std::endl;
     }
     }
     return arrout;
     return arrout;
   }
   }