/// <summary> /// 进行模版匹配 /// </summary> /// <param name="sourceImage">对应的查询(原始)图像</param> /// <param name="searchImage">对应的训练(模板)图像(宽高不得超过被查询图像)</param> /// <param name="type">匹配算法</param> /// <param name="argument">匹配参数(可选)</param> /// <returns></returns> public static TemplateMatchResult TemplateMatch(Mat sourceImage, Mat searchImage, TemplateMatchType type = TemplateMatchType.CCOEFF_NORMED, TemplateMatchArgument argument = null) { if (searchImage.Empty()) { throw new ArgumentOutOfRangeException(nameof(searchImage), "不允许使用空的查询(原始)图像进行识别"); } if (sourceImage.Empty()) { throw new ArgumentOutOfRangeException(nameof(searchImage), "不允许使用空的查询(原始)图像进行识别"); } if (sourceImage.Type() == searchImage.Type()) { return(Match.TemplateMatch.Match(sourceImage, searchImage, type, argument)); } using var sourceMat = new Mat(sourceImage.Rows, sourceImage.Cols, MatType.CV_8UC1); using var searchMat = new Mat(searchImage.Rows, searchImage.Cols, MatType.CV_8UC1); Cv2.CvtColor(sourceImage, sourceMat, ColorConversionCodes.BGR2GRAY); Cv2.CvtColor(searchImage, searchMat, ColorConversionCodes.BGR2GRAY); return(Match.TemplateMatch.Match(sourceMat, searchImage, type, argument)); }
/// <summary> /// 进行模版匹配 /// </summary> /// <param name="sourceImage">对应的查询(原始)图像</param> /// <param name="searchImage">对应的训练(模板)图像(宽高不得超过被查询图像)</param> /// <param name="type">匹配算法</param> /// <param name="argument">匹配参数(可选)</param> /// <returns></returns> public static TemplateMatchResult TemplateMatch(Bitmap sourceImage, Bitmap searchImage, TemplateMatchType type = TemplateMatchType.CCOEFF_NORMED, TemplateMatchArgument argument = null) { if (sourceImage == null) { throw new ArgumentNullException(nameof(sourceImage)); } if (searchImage == null) { throw new ArgumentNullException(nameof(searchImage)); } using var sourceMat = OpenCvSharp.Extensions.BitmapConverter.ToMat(sourceImage); using var searchMat = OpenCvSharp.Extensions.BitmapConverter.ToMat(searchImage); if (sourceMat.Width < searchMat.Width || sourceMat.Height < searchMat.Height) { throw new ArgumentException("对应的训练(模板)图像sourceImage,宽高不得超过searchImage。"); } return(TemplateMatch(sourceMat, searchMat, type, argument)); }
/// <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); }
/// <summary> /// 进行模版匹配 /// </summary> /// <param name="sourceImage">对应的查询(原始)图像</param> /// <param name="searchImage">对应的训练(模板)图像(宽高不得超过被查询图像)</param> /// <param name="type">匹配算法</param> /// <param name="argument">匹配参数(可选)</param> /// <returns></returns> public static TemplateMatchResult TemplateMatch(string sourceImage, string searchImage, TemplateMatchType type = TemplateMatchType.CCOEFF_NORMED, TemplateMatchArgument argument = null) { if (!File.Exists(sourceImage)) { throw new FileNotFoundException(sourceImage); } if (!File.Exists(searchImage)) { throw new FileNotFoundException(searchImage); } using var sourceMat = new Mat(sourceImage); using var searchMat = new Mat(searchImage); if (sourceMat.Width < searchMat.Width || sourceMat.Height < searchMat.Height) { throw new ArgumentException("对应的训练(模板)图像sourceImage,宽高不得超过searchImage。"); } return(TemplateMatch(sourceMat, searchMat, type, argument)); }
/// <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); }