From 2e12f2dfd24614acbf42120d9ba9b02142ef9c6a Mon Sep 17 00:00:00 2001
From: Bin Long <sjtubinlong@users.noreply.github.com>
Date: Tue, 25 Feb 2020 11:58:17 +0800
Subject: [PATCH] Update mask_detector.cc

---
 demo/mask_detection/cpp/mask_detector.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/mask_detection/cpp/mask_detector.cc b/demo/mask_detection/cpp/mask_detector.cc
index 06e17eba..850636fc 100644
--- a/demo/mask_detection/cpp/mask_detector.cc
+++ b/demo/mask_detection/cpp/mask_detector.cc
@@ -258,7 +258,7 @@ void MaskClassifier::Predict(std::vector<FaceResult>* faces) {
   predictor_->ZeroCopyRun();
   // Get output tensor
   auto output_names = predictor_->GetOutputNames();
-  auto out_tensor = predictor_->GetOutputTensor(output_names[1]);
+  auto out_tensor = predictor_->GetOutputTensor(output_names[0]);
   std::vector<int> output_shape = out_tensor->shape();
   // Calculate output length
   int output_size = 1;
-- 
GitLab