mirror of
https://github.com/deepinsight/insightface.git
synced 2026-01-18 15:40:18 +00:00
Added the system resource management and monitoring functions
This commit is contained in:
@@ -2,6 +2,8 @@
|
||||
#include "settings/test_settings.h"
|
||||
#include "inspireface/c_api/inspireface.h"
|
||||
#include "inspireface/herror.h"
|
||||
#include "opencv2/opencv.hpp"
|
||||
#include "unit/test_helper/test_tools.h"
|
||||
#include <cstdio>
|
||||
|
||||
TEST_CASE("test_System", "[system]") {
|
||||
@@ -37,4 +39,230 @@ TEST_CASE("test_System", "[system]") {
|
||||
ret = HFReleaseInspireFaceSession(session);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_CASE("test_SystemSessionReleaseCase", "[system]") {
|
||||
/**
|
||||
* @brief Test the release of sessions
|
||||
* @details Test the release of sessions and check the unreleased sessions count
|
||||
*/
|
||||
DRAW_SPLIT_LINE
|
||||
TEST_PRINT_OUTPUT(true);
|
||||
HResult ret;
|
||||
|
||||
SECTION("CreateSessions") {
|
||||
/**
|
||||
* @brief Create sessions
|
||||
* @details Create 10 sessions and check the unreleased sessions count
|
||||
*/
|
||||
HInt32 count;
|
||||
ret = HFDeBugGetUnreleasedSessionsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 0);
|
||||
|
||||
HInt32 createCount = 10;
|
||||
HFSession sessions[createCount];
|
||||
|
||||
for (int i = 0; i < createCount; ++i) {
|
||||
ret = HFCreateInspireFaceSessionOptional(HF_ENABLE_NONE, HF_DETECT_MODE_ALWAYS_DETECT, 3, -1, -1, &sessions[i]);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
}
|
||||
|
||||
ret = HFDeBugGetUnreleasedSessionsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == createCount);
|
||||
|
||||
HFSession sessionsGet[createCount];
|
||||
ret = HFDeBugGetUnreleasedSessions(sessionsGet, createCount);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
// The session list obtained from the api is also unordered because it is sorted internally using an unordered dictionary
|
||||
for (int i = 0; i < createCount; ++i) {
|
||||
bool found = false;
|
||||
for (int j = 0; j < createCount; ++j) {
|
||||
if (sessions[i] == sessionsGet[j]) {
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
REQUIRE(found);
|
||||
}
|
||||
}
|
||||
|
||||
SECTION("ReleaseSomeSessions") {
|
||||
/**
|
||||
* @brief Release some sessions
|
||||
* @details Release some sessions and check the unreleased sessions count
|
||||
*/
|
||||
HInt32 count;
|
||||
ret = HFDeBugGetUnreleasedSessionsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 10);
|
||||
auto createCount = count;
|
||||
|
||||
HFSession sessionsGet[createCount];
|
||||
ret = HFDeBugGetUnreleasedSessions(sessionsGet, createCount);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
|
||||
std::vector<int32_t> releaseIndex = {0, 2, 4, 6, 8};
|
||||
for (int i = 0; i < releaseIndex.size(); ++i) {
|
||||
ret = HFReleaseInspireFaceSession(sessionsGet[releaseIndex[i]]);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
}
|
||||
|
||||
ret = HFDeBugGetUnreleasedSessionsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == createCount - releaseIndex.size());
|
||||
|
||||
HFSession sessionsGet2[count];
|
||||
ret = HFDeBugGetUnreleasedSessions(sessionsGet2, count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
for (int i = 0; i < count; ++i) {
|
||||
bool found = false;
|
||||
for (int j = 0; j < releaseIndex.size(); ++j) {
|
||||
if (sessionsGet2[i] == sessionsGet[releaseIndex[j]]) {
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
REQUIRE(!found);
|
||||
}
|
||||
}
|
||||
|
||||
SECTION("ReleaseAllSessions") {
|
||||
/**
|
||||
* @brief Release all sessions
|
||||
* @details Release all sessions and check the unreleased sessions count
|
||||
*/
|
||||
HInt32 count;
|
||||
ret = HFDeBugGetUnreleasedSessionsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 5);
|
||||
|
||||
HFSession sessionsGet[count];
|
||||
ret = HFDeBugGetUnreleasedSessions(sessionsGet, count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
|
||||
for (int i = 0; i < count; ++i) {
|
||||
ret = HFReleaseInspireFaceSession(sessionsGet[i]);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
}
|
||||
|
||||
ret = HFDeBugGetUnreleasedSessionsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 0);
|
||||
}
|
||||
}
|
||||
|
||||
TEST_CASE("test_SystemStreamReleaseCase", "[system]") {
|
||||
/**
|
||||
* @brief Test the release of streams
|
||||
* @details Test the release of streams and check the unreleased streams count
|
||||
*/
|
||||
DRAW_SPLIT_LINE
|
||||
TEST_PRINT_OUTPUT(true);
|
||||
HResult ret;
|
||||
|
||||
SECTION("CreateStreams") {
|
||||
/**
|
||||
* @brief Create streams
|
||||
* @details Create 10 streams and check the unreleased streams count
|
||||
*/
|
||||
HInt32 count;
|
||||
ret = HFDeBugGetUnreleasedStreamsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 0);
|
||||
|
||||
HInt32 createCount = 10;
|
||||
HFImageStream streams[createCount];
|
||||
|
||||
for (int i = 0; i < createCount; ++i) {
|
||||
HFImageStream imgHandle;
|
||||
auto image = cv::imread(GET_DATA("data/bulk/pedestrian.png"));
|
||||
ret = CVImageToImageStream(image, imgHandle);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
streams[i] = imgHandle;
|
||||
}
|
||||
|
||||
ret = HFDeBugGetUnreleasedStreamsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == createCount);
|
||||
|
||||
HFImageStream streamsGet[createCount];
|
||||
ret = HFDeBugGetUnreleasedStreams(streamsGet, createCount);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
for (int i = 0; i < createCount; ++i) {
|
||||
bool found = false;
|
||||
for (int j = 0; j < createCount; ++j) {
|
||||
if (streams[i] == streamsGet[j]) {
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
REQUIRE(found);
|
||||
}
|
||||
}
|
||||
|
||||
SECTION("ReleaseSomeStreams") {
|
||||
/**
|
||||
* @brief Release some streams
|
||||
* @details Release some streams and check the unreleased streams count
|
||||
*/
|
||||
HInt32 count;
|
||||
ret = HFDeBugGetUnreleasedStreamsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 10);
|
||||
auto createCount = count;
|
||||
|
||||
HFImageStream streamsGet[count];
|
||||
ret = HFDeBugGetUnreleasedStreams(streamsGet, count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
|
||||
std::vector<int32_t> releaseIndex = {0, 2, 4, 6, 8};
|
||||
for (int i = 0; i < releaseIndex.size(); ++i) {
|
||||
ret = HFReleaseImageStream(streamsGet[releaseIndex[i]]);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
}
|
||||
|
||||
ret = HFDeBugGetUnreleasedStreamsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == createCount - releaseIndex.size());
|
||||
|
||||
HFImageStream streamsGet2[count];
|
||||
ret = HFDeBugGetUnreleasedStreams(streamsGet2, count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
for (int i = 0; i < count; ++i) {
|
||||
bool found = false;
|
||||
for (int j = 0; j < releaseIndex.size(); ++j) {
|
||||
if (streamsGet2[i] == streamsGet[releaseIndex[j]]) {
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
REQUIRE(!found);
|
||||
}
|
||||
}
|
||||
|
||||
SECTION("ReleaseAllStreams") {
|
||||
/**
|
||||
* @brief Release all streams
|
||||
* @details Release all streams and check the unreleased streams count
|
||||
*/
|
||||
HInt32 count;
|
||||
ret = HFDeBugGetUnreleasedStreamsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 5);
|
||||
|
||||
HFImageStream streamsGet[count];
|
||||
ret = HFDeBugGetUnreleasedStreams(streamsGet, count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
|
||||
for (int i = 0; i < count; ++i) {
|
||||
ret = HFReleaseImageStream(streamsGet[i]);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
}
|
||||
|
||||
ret = HFDeBugGetUnreleasedStreamsCount(&count);
|
||||
REQUIRE(ret == HSUCCEED);
|
||||
REQUIRE(count == 0);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user