如何减少软件 GUI 上 openCV 模板匹配的误报

我正在尝试创建一个用于游戏的任务器,就像一个机器人来执行重复的简单任务。


我正在寻找的一些元素(如文本和按钮)非常相似(见下图)。


我什至使用蒙版来获得最佳效果,模板的大小与原始图片完全相同。


所以,在游戏的主屏幕上,只有一个元素必须匹配,所以我可以决定机器人必须执行什么动作,但这些是每次匹配的等级:


img/btnCara.png 0.960


img/btnCoroa.png 0.960


img/btnFinalizar.png 0.988


img/btnRender.png 0.875


img/btnSim.png 0.997


img/btnJogar.png 0.922 << 图片上唯一的一个


img/txtEnemyChosen.png 0.953


img/txtJogarPrimeiro.png 0.945


img/txtVocePerdeu.png 0.951 格式


那么,我怎样才能减少误报并只在屏幕上找到我要找的东西呢?


我正在使用的代码在这里:


def SearchImage(img):

    partial_image = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

    partial_image= cv2.threshold(partial_image, 0, 255, cv2.THRESH_BINARY)[1]


    # get largest contour from binary image

    contours = cv2.findContours(partial_image.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    contours = contours[0] if len(contours) == 2 else contours[1]

    big_contour = max(contours, key=cv2.contourArea)


    # draw the contour of the piece outline as the mask

    mask = np.zeros((img.shape[0], img.shape[1], 3), dtype=np.uint8)

    cv2.drawContours(mask, [big_contour], 0, (255,255,255), 1)

    hh, ww = mask.shape[:2]


    # extract the template from the BGR (no alpha) piece 

    template = img[:,:,0:3]

    correlation = cv2.matchTemplate(img_np, template, cv2.TM_CCORR_NORMED, mask=mask)

    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(correlation)

    max_val_ncc = '{:.3f}'.format(max_val)

    #print("normalize_cross_correlation: " + max_val_ncc)

    xx = max_loc[0]

    yy = max_loc[1]

    #print(xx, yy)

        

    # draw template bounds and corner intersection in red onto img

    cv2.rectangle(screenshot, (xx, yy), (xx+ww, yy+hh), (0, 0, 255), 1)

    

    #return xx, yy

    return max_val_ncc

这是游戏的图形用户界面:

http://img1.mukewang.com/649144a800017a3215940937.jpg

这是我在 GUI 上寻找的元素列表

http://img3.mukewang.com/649144b40001a04416270281.jpg

如您所见,其中一些非常相似,例如这两个:

http://img.mukewang.com/649144c200014f0a03220043.jpg


青春有我
浏览 84回答 0
0回答
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python