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-23 14:46:20 +08:00
parent 445d6fc37c
commit 526bb6c2db
3 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/Matrox/template_matching.cpp
View File

@ -112,7 +112,7 @@ void TemplateMatcher::loadTemplates(const std::vector<std::string>& template_pat
}
// Search for models in the input image
void TemplateMatcher::findModels(const MIL_ID& inputImage)
void TemplateMatcher::findModels(const MIL_ID& inputImage,MIL_ID& outputImage)
{
if (!isInitialized) {
std::cerr << "Templates are not loaded. Please load templates before searching.\n";
@ -147,6 +147,13 @@ void TemplateMatcher::findModels(const MIL_ID& inputImage)
MmodGetResult(MilResult, M_DEFAULT, M_SCALE, Scale.data());
MmodGetResult(MilResult, M_DEFAULT, M_SCORE, Score.data());
// Create a binary image buffer
MbufAlloc2d(MilSystem, MbufInquire(inputImage, M_SIZE_X, M_NULL),
MbufInquire(inputImage, M_SIZE_Y, M_NULL), 1 + M_UNSIGNED,
M_IMAGE + M_PROC, &outputImage);
// Initialize the binary image to black
MbufClear(outputImage, 0);
// Display results
std::cout << "Found " << NumResults << " model(s) in " << Time * 1000.0 << " ms:\n";
std::cout << "Result Model X Position Y Position Angle Scale Score\n";
@ -155,11 +162,20 @@ void TemplateMatcher::findModels(const MIL_ID& inputImage)
<< YPosition[i] << " " << Angle[i] << " " << Scale[i]
<< " " << Score[i] << "%\n";
// Draw results
// Draw results onto the binary image
MgraColor(M_DEFAULT, 255); // White color for binary image
MmodDraw(M_DEFAULT, MilResult, outputImage, M_DRAW_EDGES + M_DRAW_POSITION, i, M_DEFAULT);
// Draw results on the graphical list for display
MgraColor(M_DEFAULT, ModelsDrawColor[Models[i]]);
MmodDraw(M_DEFAULT, MilResult, GraphicList,
M_DRAW_EDGES + M_DRAW_POSITION, i, M_DEFAULT);
}
// Display or save the binary image
MbufSave(SAVE_PATH2, outputImage);
} else {
std::cout << "No models found.\n";
}
@ -167,6 +183,7 @@ void TemplateMatcher::findModels(const MIL_ID& inputImage)
MosGetch();
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)
@ -189,10 +206,10 @@ void TemplateMatcher::LoadTemplate(TemplateMatcher& matcher, std::map<std::strin
);
}
void TemplateMatcher::FindTemplates( const MIL_ID& inputImage,const MIL_ID& outputImage,TemplateMatcher& matcher)
void TemplateMatcher::FindTemplates( const MIL_ID& inputImage, MIL_ID& outputImage,TemplateMatcher& matcher)
{
// Perform template matching
matcher.findModels(inputImage);
matcher.findModels(inputImage,outputImage);
// Notify user that matching is complete
cout << "Template matching completed.\n";

5
src/Matrox/template_matching.h
View File

@ -18,6 +18,7 @@ private:
MIL_ID MilSearchContext;
MIL_ID MilResult;
MIL_ID GraphicList;
#define SAVE_PATH2 MIL_TEXT("C:\\Users\\zjc\\Desktop\\detection.png")
std::vector<std::string> ModelImgPaths;
std::vector<MIL_INT> ModelsOffsetX;
@ -43,11 +44,11 @@ public:
const std::vector<MIL_DOUBLE>& drawColor);
// Search for models in the input image
void findModels(const MIL_ID& inputImage);
void findModels(const MIL_ID& inputImage,MIL_ID& outputImage);
void LoadTemplate(TemplateMatcher &matcher, std::map<std::string, int> &params);
void FindTemplates(const MIL_ID &inputImage, const MIL_ID &outputImage, TemplateMatcher &matcher);
void FindTemplates(const MIL_ID &inputImage, MIL_ID &outputImage, TemplateMatcher &matcher);
// Destructor

2
tests/test_template_matching.cpp
View File

@ -47,8 +47,6 @@ int main() {
std::cout << "所有颜色检测已完成并合并。按 <Enter> 退出。" << std::endl;
getchar();
MbufFree(detection_result);
MbufFree(MilImage);