NanjingForestryUniversity/cotton_color
3
1
Fork 0
mirror of https://github.com/Karllzy/cotton_color.git synced 2025-11-08 18:53:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

将加载模板的方式修改为加载配置文件

Browse Source
This commit is contained in:
zjc-zjc-123 2024-11-25 13:32:40 +08:00
parent 119f01ae88
commit fc50c3b440
3 changed files with 80 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
src/Matrox/template_matching.cpp
View File

@ -184,28 +184,74 @@ void TemplateMatcher::findModels(const MIL_ID& inputImage,MIL_ID& outputImage)
MbufFree(input_image_uint8);
}
void TemplateMatcher::LoadTemplate(TemplateMatcher& matcher, std::map<std::string, int> &params)
{
// Create a TemplateMatcher instance (consider making it static if you want to retain it between calls)
// Load template models only once
matcher.loadTemplates(
{
"C:\\Users\\zjc\\Desktop\\templates\\template1.png",
"C:\\Users\\zjc\\Desktop\\templates\\template2.png",
"C:\\Users\\zjc\\Desktop\\templates\\template3.png",
"C:\\Users\\zjc\\Desktop\\templates\\template4.png",
"C:\\Users\\zjc\\Desktop\\templates\\template5.png",
"C:\\Users\\zjc\\Desktop\\templates\\template6.png",
},
{0, 0, 0, 0,0,0}, // offsetX
{0, 0, 0, 0,0,0}, // offsetY
{100, 80, 200, 96,96,96}, // sizeX
{100, 80, 86, 96,96,96}, // sizeY
{M_COLOR_RED, M_COLOR_GREEN, M_COLOR_BLUE, M_COLOR_GREEN,M_COLOR_BLUE,M_COLOR_BLUE} // drawColor
);
std::vector<std::string> splitString(const std::string& str, char delimiter) {
std::vector<std::string> tokens;
std::stringstream ss(str);
std::string item;
while (std::getline(ss, item, delimiter)) {
tokens.push_back(item);
}
return tokens;
}
void TemplateMatcher::loadConfig(const std::string& filename,
std::vector<std::string>& template_paths,
std::vector<MIL_INT>& offsetX,
std::vector<MIL_INT>& offsetY,
std::vector<MIL_INT>& sizeX,
std::vector<MIL_INT>& sizeY,
std::vector<MIL_DOUBLE>& drawColor) {
std::ifstream file(filename);
if (!file.is_open()) {
std::cerr << "Unable to open configuration file: " << filename << std::endl;
return;
}
std::string line;
while (std::getline(file, line)) {
auto pos = line.find('=');
if (pos == std::string::npos) continue;
std::string key = line.substr(0, pos);
std::string value = line.substr(pos + 1);
auto values = splitString(value, ',');
if (key == "template_paths") {
template_paths = values;
} else if (key == "offsetX") {
for (const auto& v : values) offsetX.push_back(std::stoi(v));
} else if (key == "offsetY") {
for (const auto& v : values) offsetY.push_back(std::stoi(v));
} else if (key == "sizeX") {
for (const auto& v : values) sizeX.push_back(std::stoi(v));
} else if (key == "sizeY") {
for (const auto& v : values) sizeY.push_back(std::stoi(v));
} else if (key == "drawColor") {
for (const auto& v : values) {
if (v == "M_COLOR_RED") drawColor.push_back(M_COLOR_RED);
else if (v == "M_COLOR_GREEN") drawColor.push_back(M_COLOR_GREEN);
else if (v == "M_COLOR_BLUE") drawColor.push_back(M_COLOR_BLUE);
}
}
}
file.close();
}
void TemplateMatcher::LoadTemplate(TemplateMatcher& matcher, std::map<std::string, int>& params)
{
std::vector<std::string> template_paths;
std::vector<MIL_INT> offsetX, offsetY, sizeX, sizeY;
std::vector<MIL_DOUBLE> drawColor;
// 调用 loadConfig 并加载配置
loadConfig("C:\\Users\\zjc\\Desktop\\config\\template_config.txt",
template_paths, offsetX, offsetY, sizeX, sizeY, drawColor);
// 调用 matcher 的 loadTemplates 方法
matcher.loadTemplates(template_paths, offsetX, offsetY, sizeX, sizeY, drawColor);
}
void TemplateMatcher::FindTemplates( const MIL_ID& inputImage, MIL_ID& outputImage,TemplateMatcher& matcher)
{
// Perform template matching
@ -215,7 +261,8 @@ void TemplateMatcher::FindTemplates( const MIL_ID& inputImage, MIL_ID& outputIma
cout << "Template matching completed.\n";
}
//TODO: 1加入加载多个模板的功能 已 + 加入配置文件
//TODO: 1加入加载多个模板的功能 已 + 加入配置文件 已
//TODO: 5制作标准结构的函数，例如：matcher.findModels(MIL_ID inputImage, MIL_ID output_image, map);
////未实现，因为加载和寻找分开后，要对加载和寻找函数传入类成员，无法统一，其余可用到的参数统一，加一个类成员即可。

11
src/Matrox/template_matching.h
View File

@ -9,6 +9,9 @@
void pre_process(const MIL_ID& inputImage, MIL_ID& outputImageSuspect, const std::map<std::string, int>& params);
// void LoadTemplate(const MIL_ID &inputImage, MIL_ID &outputImage, std::map<std::string, int> &params);
extern std::vector<std::string> template_paths;
extern std::vector<MIL_INT> offsetX, offsetY, sizeX, sizeY;
extern std::vector<MIL_DOUBLE> drawColor;
class TemplateMatcher {
@ -50,7 +53,13 @@ void findModels(const MIL_ID& inputImage,MIL_ID& outputImage);
void FindTemplates(const MIL_ID &inputImage, MIL_ID &outputImage, TemplateMatcher &matcher);
void loadConfig(const std::string& filename,
std::vector<std::string>& template_paths,
std::vector<MIL_INT>& offsetX,
std::vector<MIL_INT>& offsetY,
std::vector<MIL_INT>& sizeX,
std::vector<MIL_INT>& sizeY,
std::vector<MIL_DOUBLE>& drawColor);
// Destructor
~TemplateMatcher();

3
tests/test_template_matching.cpp
View File

@ -22,7 +22,7 @@ int main() {
std::map<std::string, int> params;
read_params_from_file("C:\\Users\\zjc\\Desktop\\config\\template_config.txt", params);
read_params_from_file("C:\\Users\\zjc\\Desktop\\config\\template_color_config.txt", params);
// Initialize MIL application
MappAllocDefault(M_DEFAULT, &MilApplication, &MilSystem, &MilDisplay, M_NULL,
M_NULL);
@ -36,6 +36,7 @@ int main() {
MIL_ID detection_resize = M_NULL;
MIL_ID output_Image= M_NULL;
TemplateMatcher matcher(MilSystem, MilDisplay, params);