Skip to content

Commit 923e0b4

Browse files
committed
use separate sp and net
1 parent 349fe5a commit 923e0b4

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

visionSamples/FaceTracker/app/src/main/cpp/native-lib.cpp

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,9 @@ using anet_type = dlib::loss_metric<dlib::fc_no_bias<128, dlib::avg_pool_everyth
4848
>>>>>>>>>>>>;
4949

5050
dlib::frontal_face_detector detector = dlib::get_frontal_face_detector();
51-
dlib::shape_predictor sp;
52-
anet_type net;
51+
dlib::frontal_face_detector detector1 = dlib::get_frontal_face_detector();
52+
dlib::shape_predictor sp, sp1;
53+
anet_type net, net1;
5354

5455
//std::vector<matrix<float, 0, 1>> known_faces;
5556

@@ -174,8 +175,12 @@ Java_com_google_android_gms_samples_vision_face_facetracker_FaceTrackerActivity_
174175
//fclose(file3);
175176
dlib::deserialize("/storage/emulated/0/Download/shape_predictor_5_face_landmarks.dat")
176177
>> sp;
178+
dlib::deserialize("/storage/emulated/0/Download/shape_predictor_5_face_landmarks.dat")
179+
>> sp1;
177180
dlib::deserialize("/storage/emulated/0/Download/dlib_face_recognition_resnet_model_v1.dat")
178181
>> net;
182+
dlib::deserialize("/storage/emulated/0/Download/dlib_face_recognition_resnet_model_v1.dat")
183+
>> net1;
179184
//dlib::deserialize("/storage/emulated/0/Download/faces_linear.svm") >> df;
180185

181186
DIR *d;
@@ -200,9 +205,6 @@ Java_com_google_android_gms_samples_vision_face_facetracker_FaceTrackerActivity_
200205
matrix<float, 0, 1> face_vector;
201206
dlib::deserialize("/storage/emulated/0/Download/" + file) >> face_vector;
202207
known_faces.insert({name, face_vector});
203-
204-
//printf("%s\t",p1);
205-
//LOGI("item %s", p1);
206208
}
207209
}
208210

@@ -243,7 +245,7 @@ Java_com_google_android_gms_samples_vision_face_facetracker_FaceTrackerActivity_
243245

244246
img.set_size(infocolor.height, infocolor.width);
245247

246-
//LOGI("size w=%d h=%d", infocolor.width, infocolor.height);
248+
LOGI("size w=%d h=%d", infocolor.width, infocolor.height);
247249
for (y = 0; y < infocolor.height; y++) { //todo: performance
248250
argb *line = (argb *) pixelscolor;
249251
for (x = 0; x < infocolor.width; ++x) {
@@ -254,20 +256,20 @@ Java_com_google_android_gms_samples_vision_face_facetracker_FaceTrackerActivity_
254256
}
255257

256258
std::string returnValue = "Num Faces: ";
257-
std::vector<dlib::rectangle> dets = detector(img);
259+
std::vector<dlib::rectangle> dets = detector1(img);
258260
returnValue += std::to_string(dets.size());
259261
returnValue += ". ";
260262

261-
std::vector<matrix<rgb_pixel>> faces;
263+
std::vector<matrix<rgb_pixel>> faces;
262264
for (auto face : dets)
263265
{
264-
auto shape = sp(img, face);
266+
auto shape = sp1(img, face);
265267
matrix<rgb_pixel> face_chip;
266268
extract_image_chip(img, get_face_chip_details(shape, 150, 0.25), face_chip);
267269
faces.push_back(move(face_chip));
268270
}
269271

270-
std::vector<matrix<float, 0, 1>> face_descriptors = net(faces);
272+
std::vector<matrix<float, 0, 1>> face_descriptors = net1(faces);
271273

272274
for (size_t i = 0; i < face_descriptors.size(); ++i)
273275
{

0 commit comments

Comments
 (0)