未验证 提交 2e12f2df 编写于 作者: B Bin Long 提交者: GitHub

Update mask_detector.cc

上级 797445e5
...@@ -258,7 +258,7 @@ void MaskClassifier::Predict(std::vector<FaceResult>* faces) { ...@@ -258,7 +258,7 @@ void MaskClassifier::Predict(std::vector<FaceResult>* faces) {
predictor_->ZeroCopyRun(); predictor_->ZeroCopyRun();
// Get output tensor // Get output tensor
auto output_names = predictor_->GetOutputNames(); 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(); std::vector<int> output_shape = out_tensor->shape();
// Calculate output length // Calculate output length
int output_size = 1; int output_size = 1;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册