/// <summary>
/// 进行模版匹配
/// </summary>
/// <param name="sourceMat">对应的查询（原始）图像</param>
/// <param name="searchMat">对应的训练（模板）图像</param>
/// <param name="type">匹配算法</param>
/// <param name="argument">匹配参数（可选）</param>
/// <returns></returns>
internal static TemplateMatchResult Match(Mat sourceMat, Mat searchMat, TemplateMatchType type = TemplateMatchType.CCOEFF_NORMED, TemplateMatchArgument argument = null)
{
var matchModes = ConvertToMatchModes(type);
using var resultMat = new Mat();
resultMat.Create(sourceMat.Rows - searchMat.Rows + 1, sourceMat.Cols - searchMat.Cols + 1,
MatType.CV_32FC1);
//进行模版匹配
Cv2.MatchTemplate(sourceMat, searchMat, resultMat, matchModes);
//对结果进行归一化
Cv2.Normalize(resultMat, resultMat, 1, 0, NormTypes.MinMax, -1);
//如果没有传入匹配参数，则使用默认参数
argument ??= new TemplateMatchArgument();
var matchResult = GetMatchResult(searchMat, resultMat, matchModes, argument);
argument.OutputDebugMessage($"[TemplateMatch] [{type}] The result of the match is ({matchResult.Success}) ({matchResult.MatchItems.Count}).");
if (matchResult.Success)
{
var bestMatch = matchResult.MatchItems[0];
argument.OutputDebugMessage($"[TemplateMatch] [{type}] The center point of the best match is ({bestMatch.Point}), and the rect is {bestMatch.Rectangle}.");
}
argument.PreviewDebugTemplateMatchResult(matchResult, sourceMat);
return(matchResult);
}
/// <summary>
/// 获取匹配结果
/// </summary>
/// <param name="searchMat">对应的训练（模板）图像</param>
/// <param name="resultMat">匹配结果</param>
/// <param name="matchModes">匹配算法</param>
/// <param name="argument">匹配参数</param>
/// <returns></returns>
private static TemplateMatchResult GetMatchResult(Mat searchMat, Mat resultMat, TemplateMatchModes matchModes, TemplateMatchArgument argument)
{
var threshold = argument.Threshold;
var maxCount = argument.MaxCount;
var matchResult = new TemplateMatchResult();
while (matchResult.MatchItems.Count < maxCount)
{
double value;
Point topLeft;
Cv2.MinMaxLoc(resultMat, out var minValue, out var maxValue, out var minLocation, out var maxLocation);
if (matchModes == TemplateMatchModes.SqDiff || matchModes == TemplateMatchModes.SqDiffNormed)
{
value = minValue;
topLeft = minLocation;
}
else
{
value = maxValue;
topLeft = maxLocation;
}
if (maxValue < threshold)
{
break;
}
argument.OutputDebugMessage($"[TemplateMatch] The info of the match is ([{matchModes}][{threshold:F}][{matchResult.MatchItems.Count}]) = {value:F}.");
var matchItem = new TemplateMatchResultItem()
{
Value = value
};
var centerX = topLeft.X + (double)searchMat.Width / 2;
var centerY = topLeft.Y + (double)searchMat.Height / 2;
matchItem.Point = new System.Drawing.Point((int)centerX, (int)centerY);
matchItem.Rectangle =
new System.Drawing.Rectangle(topLeft.X, topLeft.Y, searchMat.Width, searchMat.Height);
matchResult.MatchItems.Add(matchItem);
//屏蔽已筛选区域
if (matchModes == TemplateMatchModes.SqDiff || matchModes == TemplateMatchModes.SqDiffNormed)
{
Cv2.FloodFill(resultMat, topLeft, double.MaxValue);
}
else
{
Cv2.FloodFill(resultMat, topLeft, double.MinValue);
}
}
matchResult.Success = matchResult.MatchItems.Any();
return(matchResult);
}