Leahnaya/SceneSwitcher
1
0
Fork 0
mirror of https://github.com/WarmUpTill/SceneSwitcher.git synced 2026-03-25 03:15:33 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix pattern matching by excluding non-finite results

Browse Source
This commit is contained in:
Michael Kirsch 2024-01-03 22:58:28 +01:00 committed by WarmUpTill
parent 4e84866f17
commit c341d6ebd8
1 changed files with 26 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/macro-external/video/opencv-helpers.cpp
View File

@ -22,11 +22,28 @@ PatternImageData CreatePatternData(const QImage &pattern)
return data;
}
static void invertPatternMatchResult(cv::Mat &mat)
static void preprocessPatternMatchResult(cv::Mat &mat, bool invert)
{
// Not finite values are not handled well by the cv::threshold method.
// These occur when the image is completely black, the denominator in all
// normalized algorithms approaches 0 in these cases if the alpha channel is
// used as mask.
//
// So we are clamping the values here to 0.0..1.0 and dismiss not-finite
// values.
// Invertion mode is for the TM_SQDIFF_NORMED method.
float value;
for (int r = 0; r < mat.rows; r++) {
for (int c = 0; c < mat.cols; c++) {
mat.at<float>(r, c) = 1.0 - mat.at<float>(r, c);
value = mat.at<float>(r, c);
if (invert) {
value = 1.0f - value;
}
if (!std::isfinite(value)) {
value = 0.0f;
}
mat.at<float>(r, c) =
std::fmaxf(0.0f, std::fminf(1.0f, value));
}
}
}
@ -35,6 +52,10 @@ void MatchPattern(QImage &img, const PatternImageData &patternData,
double threshold, cv::Mat &result, double *pBestFitValue,
bool useAlphaAsMask, cv::TemplateMatchModes matchMode)
{
result = cv::Mat(0, 0, CV_32F);
if (pBestFitValue) {
*pBestFitValue = std::numeric_limits<double>::signaling_NaN();
}
if (img.isNull() || patternData.rgbaPattern.empty()) {
return;
}
@ -69,9 +90,9 @@ void MatchPattern(QImage &img, const PatternImageData &patternData,
//
// For TM_CCOEFF_NORMED and TM_CCORR_NORMED a perfect match is
// represented as "1"
if (matchMode == cv::TM_SQDIFF_NORMED) {
invertPatternMatchResult(result);
}
//
// -> Invert TM_SQDIFF_NORMED in the preprocess step
preprocessPatternMatchResult(result, matchMode == cv::TM_SQDIFF_NORMED);
if (pBestFitValue) {
cv::minMaxLoc(result, nullptr, pBestFitValue);