scbc_repos/ssat_sdk/picture/object_detect.py

# -*- coding:utf-8 -*-
import os
import sys
import time
from ssat_sdk.picture.feature_detect import FeatureDetect

'''处理OpenCV图像对象'''


class ObjectDetect():
    def __init__(self):
        self.featureDetect = FeatureDetect()

    '''
    在大图bigImg中，检测iconImg是否存在，如果存在返回位置。
    :param bigPic  被检索的全图路径
    :param iconPic  需要检索的图标路径
    :param targetArea  全图区域
    :return boolean,area。boolean：True代表检测到icon，False代表没有。为True时，返回icon区域坐标，为False时，返回None。
    '''

    def detectIcon(self, bigImg, iconImg, targetArea=None):
        isFind = False
        iconArea = None
        ret = self.featureDetect.matchSingleImage(bigImg, targetArea, iconImg)
        if ret is not None and ret["tmpVal"] > 0.8:
            isFind = True
            iconArea = ret["coordinate"]
        return isFind, iconArea