From f3b98299a7e652fd3ca60fc3096b81544ac69d2b Mon Sep 17 00:00:00 2001 From: Neucrack Date: Tue, 16 Jul 2024 18:55:27 +0800 Subject: [PATCH] optimize face recognizer for dual_buff mode --- components/nn/include/maix_nn_face_recognizer.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/components/nn/include/maix_nn_face_recognizer.hpp b/components/nn/include/maix_nn_face_recognizer.hpp index e259080c..350ece50 100644 --- a/components/nn/include/maix_nn_face_recognizer.hpp +++ b/components/nn/include/maix_nn_face_recognizer.hpp @@ -319,7 +319,7 @@ namespace maix::nn image::Image *std_img = img.affine(objs->at(i).points, _std_points, _feature_input_size, _feature_input_size); // img.save("/root/test0.jpg"); // std_img->save("/root/test.jpg"); - tensor::Tensors *outputs = _model_feature->forward_image(*std_img, this->mean_feature, this->scale_feature, fit, false); + tensor::Tensors *outputs = _model_feature->forward_image(*std_img, this->mean_feature, this->scale_feature, fit, false, get_feature || get_face); if (!outputs) // not ready for dual_buff mode { delete std_img;