当前位置：网站首页>AI 之 OpenCvSharp 大图找小图（案例版）

AI 之 OpenCvSharp 大图找小图（案例版）

2022-07-17 01:57:00 【dotNET跨平台】

要说跟AI扯上啥关系，估计只有库本身了，但是，这个大图搜小图功能还是不错的，有喜欢的可以试试。我主要用的场景是，具体的某个界面为大图，界面的某个图标为小图，或者，整个桌面为大图，某个小图标为小图这样来确定某个插件或者功能的位置。

之前也写了一下基于C#代码的找图，但是感觉效率不高，这次做个基于OpenCvSharp的找图功能，有喜欢的可以尝试下。

创建一个Winfrom项目

因为项目示例简单，但是，又需要获取桌面大小，得创建这样的项目。

要添加Nuget包

Install-Package OpenCvSharp4.Windows -Version 4.6.0.20220608

查找主要业务代码

public static Rectangle Find(Image sourceImage, Image matchImage, double threshold = 0.8)
{
    var refMat = Mat.FromImageData(ImageHelper.ImageToBytes(sourceImage), ImreadModes.AnyColor);//大图
    var tplMat = Mat.FromImageData(ImageHelper.ImageToBytes(matchImage), ImreadModes.AnyColor);//小图
    using (Mat res = new Mat(refMat.Rows - tplMat.Rows + 1, refMat.Cols - tplMat.Cols + 1, MatType.CV_32FC1))
    {
        Mat gref = refMat.CvtColor(ColorConversionCodes.BGR2GRAY);
        Mat gtpl = tplMat.CvtColor(ColorConversionCodes.BGR2GRAY);

        Cv2.MatchTemplate(gref, gtpl, res, TemplateMatchModes.CCoeffNormed);
        Cv2.Threshold(res, res, 0.8, 1.0, ThresholdTypes.Tozero);

        double minval, maxval;
        Point minloc, maxloc;

        Cv2.MinMaxLoc(res, out minval, out maxval, out minloc, out maxloc);

        if (maxval >= threshold)
        {
            return new Rectangle(maxloc.X, maxloc.Y, tplMat.Width, tplMat.Height);
        }
        return Rectangle.Empty;
    }
}

可以根据相似度来实现搜索，其实如果直接屏幕截图的话，相似度都是0.99，还挺不错的。

多图结果返回的话，想研究的大佬，可以研究研究。