Code: Select all
#include
#include
#include
#include
#include
namespace fs = std::filesystem;
using namespace cv;
using namespace std;
// Non-Maximum Suppression (NMS)
void nonMaxSuppression(vector& detectedRegions, double overlapThreshold) {
vector indices;
sort(detectedRegions.begin(), detectedRegions.end(), [](const Rect& a, const Rect& b) {
return a.area() > b.area();
});
for (size_t i = 0; i < detectedRegions.size(); ++i) {
bool keep = true;
for (size_t j = 0; j < i; ++j) {
float intersectionArea = (detectedRegions[i] & detectedRegions[j]).area();
float unionArea = detectedRegions[i].area() + detectedRegions[j].area() - intersectionArea;
if ((intersectionArea / unionArea) > overlapThreshold) {
keep = false;
break;
}
}
if (keep) {
indices.push_back(i);
}
}
vector filteredDetections;
for (int index : indices) {
filteredDetections.push_back(detectedRegions[index]);
}
detectedRegions = filteredDetections;
}
// template matching
void detectBlocks(const string& diagramPath, const string& templateFolder, ofstream& outputFile) {
Mat inputImage = imread(diagramPath, IMREAD_GRAYSCALE);
if (inputImage.empty()) {
cout