Files
insightface/cpp-package/inspireface/cpp/test/settings/test_settings.cpp
2024-05-02 01:27:29 +08:00

61 lines
1.5 KiB
C++

//
// Created by Tunm-Air13 on 2023/5/24.
//
#include "test_settings.h"
std::string getTestDataDir() {
return Enviro::getInstance().getTestResDir();
}
std::string getTestData(const std::string& name) {
return getTestDataDir() + "/" + name;
}
std::string getTestSaveDir() {
return getTestData("save");
}
std::string getTestSaveData(const std::string& name) {
return getTestSaveDir() + "/" + name;
}
std::string getTestModelsFile() {
std::string path = "pack/";
path = path + TEST_MODEL_FILE;
return getTestData(path);
}
std::string getTestLFWFunneledTxt() {
return getTestData(TEST_LFW_FUNNELED_TXT);
}
std::string getTestLFWFunneledEvaTxt() {
return getLFWFunneledDir() + "/" + TEST_LFW_EVALUATION_TXT;
}
std::string getLFWFunneledDir() {
const char* testLfwFunneledDir = LFW_FUNNELED_DIR;
std::string lfwFunneledPath;
if (testLfwFunneledDir != nullptr && std::string(testLfwFunneledDir) != "") {
lfwFunneledPath = testLfwFunneledDir;
} else {
const char* lfwFunneled = std::getenv("LFW_FUNNELED_DIR");
if (lfwFunneled != nullptr) {
lfwFunneledPath = lfwFunneled;
}
}
if (lfwFunneledPath.empty()) {
spdlog::warn("lfw funneled dir is empty!");
}
return lfwFunneledPath;
}
std::string getBenchmarkRecordFile() {
return getTestSaveData(TEST_BENCHMARK_RECORD);
}
std::string getEvaluationRecordFile() {
return getTestSaveData(TEST_EVALUATION_RECORD);
}