Method detect_icons

utils/icon_detection.py:406–492

检测图像中的图标 Args: image: 目标图像（numpy数组或文件路径） icon_names: 要检测的图标名称列表 app_id: 应用ID，用于确定图标搜索路径 threshold: 相似度阈值 match_mode: 匹配模式，'any'表示匹配任意一个，'all'表示必须匹配所有 Returns: 检测结果字典，包含成功状态、匹配的图标、详细

(self, 
                    image: Union[np.ndarray, str],
                    icon_names: List[str],
                    app_id: Optional[str] = None,
                    threshold: Optional[float] = None,
                    match_mode: str = 'any')

        return str(Path(__file__).parent.parent)
    
    def detect_icons(self, 
                    image: Union[np.ndarray, str],
                    icon_names: List[str],
                    app_id: Optional[str] = None,
                    threshold: Optional[float] = None,
                    match_mode: str = 'any') -> Dict[str, Any]:
        """
        检测图像中的图标
        
        Args:
            image: 目标图像（numpy数组或文件路径）
            icon_names: 要检测的图标名称列表
            app_id: 应用ID，用于确定图标搜索路径
            threshold: 相似度阈值
            match_mode: 匹配模式，'any'表示匹配任意一个，'all'表示必须匹配所有
            
        Returns:
            检测结果字典，包含成功状态、匹配的图标、详细结果等
        """
        logger.debug(f"开始图标检测，图标列表: {icon_names}, 匹配模式: {match_mode}")
        
        result = {
            'success': False,
            'matched_icons': [],
            'unmatched_icons': [],
            'details': {},
            'total_matches': 0,
            'match_mode': match_mode
        }
        
        # 预处理图像
        processed_image = self._preprocess_image(image)
        if processed_image is None:
            result['error'] = '无法处理输入图像'
            return result
        
        # 逐个检测图标
        for icon_name in icon_names:
            # 解析图标路径
            icon_path = self.path_resolver.resolve_icon_path(icon_name, app_id)
            if icon_path is None:
                result['unmatched_icons'].append(icon_name)
                result['details'][icon_name] = {
                    'found': False,
                    'error': '图标文件未找到',
                    'matches': []
                }
                continue
            
            # 执行检测
            matches = self.detector.detect_icon(
                processed_image, 
                icon_path, 
                threshold
            )
            
            # 记录结果
            is_found = len(matches) > 0