mirror of
https://github.com/deepinsight/insightface.git
synced 2025-12-30 16:12:27 +00:00
61 lines
1.5 KiB
C++
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);
|
|
} |