o äÚižã@s>dZddlZddlmZmZmZddlZGdd„dejƒZ dS)u< 检测器基类 å®šä¹‰æ‰€æœ‰æ£€æµ‹å™¨å¿…é¡»å®žçŽ°çš„æŽ¥å£ éN)ÚDictÚAnyÚOptionalc @s,eZdZdZd$dededefdd„Zejd%d eee fd e fd d „ƒZ ejd%d e j deee fd eee ffdd„ƒZejd eee ffdd„ƒZd eeeffdd„Zd%ded e j deee fd eee ffdd„Zd&dd„Zd eeeffdd„Zdeee fd e fdd„Zdd„Zd d!„Zd"d#„ZdS)'Ú BaseDetectoru9检测器基类，所有具体检测器必须继承此类ÚÚnameÚversionÚ descriptioncCs"||_||_||_d|_i|_dS)u· åˆå§‹åŒ–æ£€æµ‹å™¨åŸºæœ¬ä¿¡æ¯ Args: name: 检测器å称 version: 检测器版本 description: 检测器æè¿° FN)rrr Ú initializedÚ parameters)Úselfrrr ©r õ7D:\pyccd\æžè€³ç¿»æŠ˜-拆分\detectors\base_detector.pyÚ__init__s   zBaseDetector.__init__NÚ config_paramsÚreturncCódS)u åˆå§‹åŒ–检测器，加载模型和é…ç½® Args: config_params: é…ç½®å‚数字典，如果为None则使用默认é…ç½® Returns: bool: åˆå§‹åŒ–æˆåŠŸè¿”回True，失败返回False Nr )r rr r rÚ initializes zBaseDetector.initializeÚimageÚparamscCr)u 执行检测 Args: image: è¾“å…¥å›¾åƒ (numpy array) params: 检测å‚数，如果为None则使用默认å‚æ•° Returns: Dict: 检测结果，必须包å«ä»¥ä¸‹å­—段： { 'success': bool, # 检测是å¦æˆåŠŸ 'fold_result': int, # 翻折状æ€ç  'errors': List[str], # é”™è¯¯ä¿¡æ¯ 'messages': List[str], # æç¤ºä¿¡æ¯ 'statistics': Dict, # ç»Ÿè®¡ä¿¡æ¯ 'details': Dict, # 详细结果 'draw_items': Dict, # 绘图项（ä¿æŒåŽŸæœ‰æ ¼å¼ï¼‰ 'processed_image': np.ndarray # 处ç†åŽçš„å›¾åƒ } Nr )r rrr r rÚdetect*szBaseDetector.detectcCr)ui 获å–默认å‚æ•°é…ç½® Returns: Dict: 默认å‚æ•°å­—å…¸ Nr ©r r r rÚget_default_parametersBsz#BaseDetector.get_default_parameterscCsddiS)u‘ 获å–检测器å¯ç”¨çš„检测函数列表 Returns: Dict[str, str]: 函数å称到æ述的映射 ru主检测函数r rr r rÚget_available_functionsLs ÿz$BaseDetector.get_available_functionsÚ function_namecCs"|dkr | ||¡Std|›ƒ‚)uö 调用指定的检测函数 Args: function_name: è¦è°ƒç”¨çš„函数å称 image: è¾“å…¥å›¾åƒ params: 检测å‚æ•° Returns: Dict: 检测结果 ru未知的函数å称: )rÚ ValueError)r rrrr r rÚ call_functionXs zBaseDetector.call_functioncCr)uœ 预热检测器（å¯é€‰æ–¹æ³•ï¼‰ å­ç±»å¯ä»¥é‡å†™æ­¤æ–¹æ³•æ¥å®žçŽ°æ¨¡åž‹é¢„热逻辑 默认实现为空æ“作 Nr rr r rÚwarm_upiszBaseDetector.warm_upcCs|j|j|j|jdœS)u„ 获å–æ£€æµ‹å™¨ä¿¡æ¯ Returns: Dict: 包å«å称ã€ç‰ˆæœ¬ã€æ述等信æ¯çš„å­—å…¸ ©rrr r rrr r rÚget_infors üzBaseDetector.get_infocCs t|tƒS)u¦ 验è¯å‚数有效性 Args: params: å‚æ•°å­—å…¸ Returns: bool: å‚数有效返回True )Ú isinstanceÚdict)r rr r rÚvalidate_parameters€s z BaseDetector.validate_parameterscCsd|_i|_dS)u0 清ç†èµ„æºï¼Œé‡Šæ”¾å†…存等 FN)r r rr r rÚcleanups zBaseDetector.cleanupcCs|j›d|j›S)Nz v)rrrr r rÚ__str__”szBaseDetector.__str__cCs d|jj›d|j›d|j›dS)Nú<ú(z, z)>)Ú __class__Ú__name__rrrr r rÚ__repr__—s zBaseDetector.__repr__)r)N)rN)r(Ú __module__Ú __qualname__Ú__doc__ÚstrrÚabcÚabstractmethodrrÚboolrÚnpZndarrayrrrrrrr"r#r$r)r r r rr s" , .    r) r,r.ÚtypingrrrZnumpyr1ÚABCrr r r rÚs