vcc_gnd
/
esp-who
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

⚡ can delete ID when no face detected

pull/216/head
yehangyang 2022-03-16 14:49:55 +08:00
parent 7199eb1619
commit 91aae97252
1 changed files with 20 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
examples/esp32-s3-eye/main/src/app_face.cpp
View File

@ -154,16 +154,17 @@ static void task(AppFace *self)
draw_detection_result((uint16_t *)frame->buf, frame->height, frame->width, detect_results); draw_detection_result((uint16_t *)frame->buf, frame->height, frame->width, detect_results);
} }
if (self->state && detect_results.size() == 1) if (self->state)
{ {
switch (self->state) if (detect_results.size() == 1)
{
if (self->state == FACE_ENROLL)
{ {
case FACE_ENROLL:
self->recognizer->enroll_id((uint16_t *)frame->buf, {(int)frame->height, (int)frame->width, 3}, detect_results.front().keypoint, "", true); self->recognizer->enroll_id((uint16_t *)frame->buf, {(int)frame->height, (int)frame->width, 3}, detect_results.front().keypoint, "", true);
ESP_LOGI(TAG, "Enroll ID %d", self->recognizer->get_enrolled_ids().back().id); ESP_LOGI(TAG, "Enroll ID %d", self->recognizer->get_enrolled_ids().back().id);
break; }
else if (self->state == FACE_RECOGNIZE)
case FACE_RECOGNIZE: {
self->recognize_result = self->recognizer->recognize((uint16_t *)frame->buf, {(int)frame->height, (int)frame->width, 3}, detect_results.front().keypoint); self->recognize_result = self->recognizer->recognize((uint16_t *)frame->buf, {(int)frame->height, (int)frame->width, 3}, detect_results.front().keypoint);
// print_detection_result(detect_results); // print_detection_result(detect_results);
ESP_LOGD(TAG, "Similarity: %f", self->recognize_result.similarity); ESP_LOGD(TAG, "Similarity: %f", self->recognize_result.similarity);
@ -171,16 +172,14 @@ static void task(AppFace *self)
ESP_LOGI(TAG, "Match ID: %d", self->recognize_result.id); ESP_LOGI(TAG, "Match ID: %d", self->recognize_result.id);
else else
ESP_LOGI(TAG, "Match ID: %d", self->recognize_result.id); ESP_LOGI(TAG, "Match ID: %d", self->recognize_result.id);
break; }
}
case FACE_DELETE: if (self->state == FACE_DELETE)
{
vTaskDelay(10); vTaskDelay(10);
self->recognizer->delete_id(true); self->recognizer->delete_id(true);
ESP_LOGI(TAG, "%d IDs left", self->recognizer->get_enrolled_id_num()); ESP_LOGI(TAG, "%d IDs left", self->recognizer->get_enrolled_id_num());
break;
default:
break;
} }
self->state_previous = self->state; self->state_previous = self->state;