diff --git a/components/modules/ai/who_human_face_detection.cpp b/components/modules/ai/who_human_face_detection.cpp new file mode 100644 index 0000000..419a1ee --- /dev/null +++ b/components/modules/ai/who_human_face_detection.cpp @@ -0,0 +1,99 @@ +#include "who_human_face_detection.hpp" + +#include "esp_log.h" +#include "esp_camera.h" + +#include "dl_image.hpp" +#include "human_face_detect_msr01.hpp" +#include "human_face_detect_mnp01.hpp" + +#include "who_ai_utils.hpp" + +#define TWO_STAGE_ON 1 + +static const char *TAG = "human_face_detection"; + +static QueueHandle_t xQueueFrameI = NULL; +static QueueHandle_t xQueueEvent = NULL; +static QueueHandle_t xQueueFrameO = NULL; +static QueueHandle_t xQueueResult = NULL; + +static bool gEvent = true; +static bool gReturnFB = true; + + +static void task_process_handler(void *arg) +{ + camera_fb_t *frame = NULL; + HumanFaceDetectMSR01 detector(0.3F, 0.3F, 10, 0.3F); +#if TWO_STAGE_ON + HumanFaceDetectMNP01 detector2(0.4F, 0.3F, 10); +#endif + + while (true) + { + if (gEvent) + { + bool is_detected = false; + if (xQueueReceive(xQueueFrameI, &frame, portMAX_DELAY)) + { +#if TWO_STAGE_ON + std::list &detect_candidates = detector.infer((uint16_t *)frame->buf, {(int)frame->height, (int)frame->width, 3}); + std::list &detect_results = detector2.infer((uint16_t *)frame->buf, {(int)frame->height, (int)frame->width, 3}, detect_candidates); +#else + std::list &detect_results = detector.infer((uint16_t *)frame->buf, {(int)frame->height, (int)frame->width, 3}); +#endif + + if (detect_results.size() > 0) + { + draw_detection_result((uint16_t *)frame->buf, frame->height, frame->width, detect_results); + print_detection_result(detect_results); + is_detected = true; + } + } + + if (xQueueFrameO) + { + xQueueSend(xQueueFrameO, &frame, portMAX_DELAY); + } + else if (gReturnFB) + { + esp_camera_fb_return(frame); + } + else + { + free(frame); + } + + if (xQueueResult) + { + xQueueSend(xQueueResult, &is_detected, portMAX_DELAY); + } + } + } +} + +static void task_event_handler(void *arg) +{ + while (true) + { + xQueueReceive(xQueueEvent, &(gEvent), portMAX_DELAY); + } +} + +void register_human_face_detection(const QueueHandle_t frame_i, + const QueueHandle_t event, + const QueueHandle_t result, + const QueueHandle_t frame_o, + const bool camera_fb_return) +{ + xQueueFrameI = frame_i; + xQueueFrameO = frame_o; + xQueueEvent = event; + xQueueResult = result; + gReturnFB = camera_fb_return; + + xTaskCreatePinnedToCore(task_process_handler, TAG, 4 * 1024, NULL, 5, NULL, 1); + if (xQueueEvent) + xTaskCreatePinnedToCore(task_event_handler, TAG, 4 * 1024, NULL, 5, NULL, 1); +} diff --git a/components/modules/ai/who_human_face_detection.hpp b/components/modules/ai/who_human_face_detection.hpp new file mode 100644 index 0000000..c3b8b4f --- /dev/null +++ b/components/modules/ai/who_human_face_detection.hpp @@ -0,0 +1,12 @@ +#pragma once + +#include "freertos/FreeRTOS.h" +#include "freertos/queue.h" +#include "freertos/task.h" +#include "freertos/semphr.h" + +void register_human_face_detection(QueueHandle_t frame_i, + QueueHandle_t event, + QueueHandle_t result, + QueueHandle_t frame_o = NULL, + const bool camera_fb_return = false);