1- /* **************************************************************************
2- *
3- * Copyright (c) 2021 Baidu.com, Inc. All Rights Reserved
4- *
5- **************************************************************************/
6-
7- /* *
8- * @author Baidu
9- * @brief demo_image_inference
10- *
11- **/
1+ // Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
2+ //
3+ // Licensed under the Apache License, Version 2.0 (the "License");
4+ // you may not use this file except in compliance with the License.
5+ // You may obtain a copy of the License at
6+ //
7+ // http://www.apache.org/licenses/LICENSE-2.0
8+ //
9+ // Unless required by applicable law or agreed to in writing, software
10+ // distributed under the License is distributed on an "AS IS" BASIS,
11+ // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ // See the License for the specific language governing permissions and
13+ // limitations under the License.
14+
1215#include " fastdeploy/vision.h"
1316
1417void CpuInfer (const std::string &model_file, const std::string ¶ms_file,
@@ -28,8 +31,8 @@ void CpuInfer(const std::string &model_file, const std::string ¶ms_file,
2831 fastdeploy::vision::FaceRecognitionResult res1;
2932 fastdeploy::vision::FaceRecognitionResult res2;
3033
31- if ((!model.Predict (& face0, &res0)) || (!model.Predict (& face1, &res1)) ||
32- (!model.Predict (& face2, &res2))) {
34+ if ((!model.Predict (face0, &res0)) || (!model.Predict (face1, &res1)) ||
35+ (!model.Predict (face2, &res2))) {
3336 std::cerr << " Prediction Failed."
3437 }
3538
@@ -40,9 +43,11 @@ void CpuInfer(const std::string &model_file, const std::string ¶ms_file,
4043 std::cout << " --- [Face 2]:" Str ();
4144
4245 float cosine01 = fastdeploy::vision::utils::CosineSimilarity (
43- res0.embedding , res1.embedding , model.l2_normalize );
46+ res0.embedding , res1.embedding ,
47+ model.GetPostprocessor ().GetL2Normalize ());
4448 float cosine02 = fastdeploy::vision::utils::CosineSimilarity (
45- res0.embedding , res2.embedding , model.l2_normalize );
49+ res0.embedding , res2.embedding ,
50+ model.GetPostprocessor ().GetL2Normalize ());
4651 std::cout << " Detect Done! Cosine 01: "
4752 << " , Cosine 02:"
4853}
@@ -65,8 +70,8 @@ void XpuInfer(const std::string &model_file, const std::string ¶ms_file,
6570 fastdeploy::vision::FaceRecognitionResult res1;
6671 fastdeploy::vision::FaceRecognitionResult res2;
6772
68- if ((!model.Predict (& face0, &res0)) || (!model.Predict (& face1, &res1)) ||
69- (!model.Predict (& face2, &res2))) {
73+ if ((!model.Predict (face0, &res0)) || (!model.Predict (face1, &res1)) ||
74+ (!model.Predict (face2, &res2))) {
7075 std::cerr << " Prediction Failed."
7176 }
7277
@@ -77,9 +82,11 @@ void XpuInfer(const std::string &model_file, const std::string ¶ms_file,
7782 std::cout << " --- [Face 2]:" Str ();
7883
7984 float cosine01 = fastdeploy::vision::utils::CosineSimilarity (
80- res0.embedding , res1.embedding , model.l2_normalize );
85+ res0.embedding , res1.embedding ,
86+ model.GetPostprocessor ().GetL2Normalize ());
8187 float cosine02 = fastdeploy::vision::utils::CosineSimilarity (
82- res0.embedding , res2.embedding , model.l2_normalize );
88+ res0.embedding , res2.embedding ,
89+ model.GetPostprocessor ().GetL2Normalize ());
8390 std::cout << " Detect Done! Cosine 01: "
8491 << " , Cosine 02:"
8592}
@@ -103,8 +110,8 @@ void GpuInfer(const std::string &model_file, const std::string ¶ms_file,
103110 fastdeploy::vision::FaceRecognitionResult res1;
104111 fastdeploy::vision::FaceRecognitionResult res2;
105112
106- if ((!model.Predict (& face0, &res0)) || (!model.Predict (& face1, &res1)) ||
107- (!model.Predict (& face2, &res2))) {
113+ if ((!model.Predict (face0, &res0)) || (!model.Predict (face1, &res1)) ||
114+ (!model.Predict (face2, &res2))) {
108115 std::cerr << " Prediction Failed."
109116 }
110117
@@ -115,9 +122,11 @@ void GpuInfer(const std::string &model_file, const std::string ¶ms_file,
115122 std::cout << " --- [Face 2]:" Str ();
116123
117124 float cosine01 = fastdeploy::vision::utils::CosineSimilarity (
118- res0.embedding , res1.embedding , model.l2_normalize );
125+ res0.embedding , res1.embedding ,
126+ model.GetPostprocessor ().GetL2Normalize ());
119127 float cosine02 = fastdeploy::vision::utils::CosineSimilarity (
120- res0.embedding , res2.embedding , model.l2_normalize );
128+ res0.embedding , res2.embedding ,
129+ model.GetPostprocessor ().GetL2Normalize ());
121130 std::cout << " Detect Done! Cosine 01: "
122131 << " , Cosine 02:"
123132}
@@ -143,8 +152,8 @@ void TrtInfer(const std::string &model_file, const std::string ¶ms_file,
143152 fastdeploy::vision::FaceRecognitionResult res1;
144153 fastdeploy::vision::FaceRecognitionResult res2;
145154
146- if ((!model.Predict (& face0, &res0)) || (!model.Predict (& face1, &res1)) ||
147- (!model.Predict (& face2, &res2))) {
155+ if ((!model.Predict (face0, &res0)) || (!model.Predict (face1, &res1)) ||
156+ (!model.Predict (face2, &res2))) {
148157 std::cerr << " Prediction Failed."
149158 }
150159
@@ -155,9 +164,11 @@ void TrtInfer(const std::string &model_file, const std::string ¶ms_file,
155164 std::cout << " --- [Face 2]:" Str ();
156165
157166 float cosine01 = fastdeploy::vision::utils::CosineSimilarity (
158- res0.embedding , res1.embedding , model.l2_normalize );
167+ res0.embedding , res1.embedding ,
168+ model.GetPostprocessor ().GetL2Normalize ());
159169 float cosine02 = fastdeploy::vision::utils::CosineSimilarity (
160- res0.embedding , res2.embedding , model.l2_normalize );
170+ res0.embedding , res2.embedding ,
171+ model.GetPostprocessor ().GetL2Normalize ());
161172 std::cout << " Detect Done! Cosine 01: "
162173 << " , Cosine 02:"
163174}
0 commit comments