o i* @sHdZddlZddlZddlZddlmZmZmZm Z m Z ddl Z ddl Z ddl Z ddlZddlZddlmZe jeZe jeZeejvrMejeddlmZzddlmZddlmZdd lmZWn.eyZz"e d eGd d d Z!e!ZGd ddZ"e"Ze"ZWYdZ[ndZ[wwddZ#GdddeZ$dS)u[ 极耳翻折检测器实现 从原始ear_fold_detection_plugin.py中提取的检测逻辑 N)DictAnyListTupleOptional)datetime) BaseDetector)yolo_model_loder)yolo_cls_model_loader)utilsu ⚠️ 无法导入utils模块: c@seZdZeddZdS) TempUtilsc Csz|durtdtjdtjdWS|jdd\}}t|dt|dt|dt|d f\}}}}td t||d }td t||d }t|||}t|||} ||ks_|| krtd |d |d|d|d|d| tjtd |td |dftjdWS||| ||f} | jd |ks| jd |krtj||dftjd} | jdd\} } | | d| d| f<| WS| WSty}z td|tj| d d| dddftjdWYd}~Sd}~ww)u9从图像中根据AOI提取子图像，包含边界检查Nu错误：输入图像为None)drdtypexywidthheightrru警告：无效AOI区域 x=, y=, w=, h=u, 图像尺寸=ru图像提取失败: r) printnpzerosuint8shapeintmaxmin Exceptionget)imageaoiZimg_hZimg_wrrwhx2y2resultpaddedZresult_hZresult_wer-;D:\pyccd\极耳翻折-拆分\detectors\ear_fold_detector.pyget_cv_img_fronm_aoi"s24,$2zTempUtils.get_cv_img_fronm_aoiN)__name__ __module__ __qualname__ staticmethodr/r-r-r-r.r !sr c@seZdZddZddZdS)TempModelLoadercCs d|_dS)Nu模型加载器未正确导入)model_load_errorselfr-r-r.__init__Js zTempModelLoader.__init__cCstd|dS)Nu⚠️ 临时模型加载器: F)r)r7pathdevicer-r-r.load_from_path_with_deviceMsz*TempModelLoader.load_from_path_with_deviceN)r0r1r2r8r;r-r-r-r.r4Is r4c Cszl|\}}|\}}|\}}||krd||} |} | | fWS||kr/|} d||} | | fWS||} ||} ||||} | | | | }|dkrN|WSd| || || |}|| |} || |} | | fWSty}ztd||WYd}~Sd}~ww)u 计算一个点关于一条直线的镜像点 Args: point: 要镜像的点 (x, y) line_point1: 直线上的第一个点 (x1, y1) line_point2: 直线上的第二个点 (x2, y2) Returns: tuple: 镜像后的点坐标 (x', y') rru镜像点计算异常: N)r"r)pointZ line_point1Z line_point2pxpyx1y1r(r)Zmirror_xZmirror_yabc denominatorZfactorr,r-r-r.mirror_point_across_lineUs6        rEc seZdZdZfddZdefddZd+deeefde fd d Z d d Z d+d e j deeefdeeeffddZddZd e j deeefdeeeffddZdeeeffddZd+ded e j deeefdeeeffddZd+d e j deeefdeeeffddZd,dd Zd e j deeefdeeeffd!d"Zd#d$Zdeeeffd%d&Zd-d'd(Zfd)d*ZZS).EarFoldDetector极耳翻折检测器cs^tjddddd|_d|_d|_t|_t|_t|_ d|_ d|_ d|_ d|_ dS)NrGz1.0.0u+基于YOLO的极耳翻折单图检测算法)nameversion descriptionzcuda:0r)superr8 cut_model detect_model gm_cls_model threadingLockcut_model_lockdetect_model_lockgm_cls_model_lock cuda_name run_countZ error_count cached_paramsr6 __class__r-r.r8s    zEarFoldDetector.__init__returncCsdS)u'返回当前检测器的配置文件名zear_fold_detection_params.jsonr-r6r-r-r.get_config_filenamesz#EarFoldDetector.get_config_filenameN config_paramsc Csz+td|dur ||_n||_t|_t|_t|_|tdd|_ WdSt yJ}ztdt |d|_ WYd}~dSd}~ww)u 初始化检测器，加载模型和配置 Args: config_params: 配置参数字典，如果为None则使用默认配置 u&🚀 初始化极耳翻折检测器...Nu✅ 检测器初始化成功Tu❌ 检测器初始化失败: F) r parametersget_default_parametersr rLrMr rN_warm_up_models initializedr"str)r7r[r,r-r-r. initializes$ zEarFoldDetector.initializec CsTz td|jdi}|jdi}|jdi}|jr|drz]|dd}|dd}|rtd |d |d |j||jrvtjjd d ||dftj d}|j |jj ||j|ddWdn1slwYtdn td|jj Wnt y}z td|WYd}~nd}~ww|jr|drz|dd}|dd} |dd} |dd} | rhtd|d |d |j| |jr_dddddd d!} |j| |d"i} | d#d}||j_| |j_|r,| d$d%| d&d%| d'd%| d(d%| d)d%| d*d%d+}|D] \}}|j||qtjjd d ||dftj d}|j|jj ||j| |dd,Wdn 1sUwYtd-n td.|jj Wnt y}z td/|WYd}~nd}~ww|jr|drz_|dd0}|dd}|rtd1|d |d |j||jrtjjd d ||dftj d}|j|jj||j|d2Wdn 1swYtd3n td4|jj Wnt y}z td5|WYd}~nd}~wwtd6WdSt y)}ztd7|WYd}~dSd}~ww)8ui 预热模型：使用虚拟数据进行一次推理，确保模型完全加载并优化 u🔥 开始模型预热...极耳裁切模型参数极耳翻折检测模型参数隔膜检测模型参数 模型路径 输入尺寸u 🔥 预热裁切模型 (rz)...rrrF)r:imgszverboseNu ✅ 裁切模型预热完成u% ⚠️ 裁切模型加载失败: u" ❌ 裁切模型预热失败: 置信度阈值皙? IOU阈值?u 🔥 预热检测模型 (JEjezaGMqiaoqigemofencengZclass1Zclass3Zclass4class0Zclass2Zclass5独立置信度参数启用独立置信度极耳置信度皙?折痕置信度月牙置信度极耳内插置信度极耳内嵌置信度分层置信度rprqrrrsrtrur:Ziourjrku ✅ 检测模型预热完成u% ⚠️ 检测模型加载失败: u" ❌ 检测模型预热失败: u! 🔥 预热隔膜分类模型 ()r:rju& ✅ 隔膜分类模型预热完成u+ ⚠️ 隔膜分类模型加载失败: u( ❌ 隔膜分类模型预热失败: u🔥 模型预热完成！u❌ 模型预热异常: )rr\r#rLr;rTrrandomZrandintrrQ predictExr5r"rMset_class_namesuse_independent_conf global_confitemsset_class_confidencerRrNload_from_pathrSpredict)r7Zcut_model_paramsZdetect_model_paramsgm_cls_model_paramsZ cut_imgszcut_model_pathZ dummy_imager,Z detect_imgszZdetect_model_pathZdetect_confidenceZ detect_iouusr_class_nameindependent_confindependent_conf_enabledindependent_conf_values class_name conf_valueZgm_imgszZ gm_model_pathr-r-r.r^s                     zEarFoldDetector._warm_up_modelsr$paramsc CsX|jsddiStd|dur|n|j}zkt}|||}i}d|vr,d|d<|WS|dg|d<|di|d<|d d |d <|jdd \}}|d i} | d d} || } || } ||dg| | |d<t} td| |dd|WSt y}z tdt |ddl }| ddt |iWYd}~Sd}~ww)u 执行检测（仅翻折检测） 按照run_all的输出格式设计，主要输出标注相关信息 错误信息检测器未初始化u 🔍 开始极耳翻折检测...N 异常结束推理异常结束cbv1_text_items result_info结果NGrrcrfrg draw_itemsdraw_items_original检测耗时: .2f秒u❌ 检测异常: ru检测异常: ) r_rr\time perf_counter_run_detectionr#r_scale_draw_items_to_originalr"r` traceback print_exc)r7r$rdetection_params start_timeZ raw_resultr*original_heightoriginal_width model_paramsZmodel_input_sizescale_xscale_yend_timer,rr-r-r.detectAsD   zEarFoldDetector.detectc Csi}|s|St|D]\}}|}|ddkrPd|vrOd|vrOt|dd|t|dd|d|d<t|dd|t|dd|d|d<n|ddkrsd |vrrt|d d|t|d d|d|d <nc|dd krd |vrg}|d D]} |t| d|t| d|dq||d <n6|dd krd |vrt|d d|t|d d|d|d <d|vr||d} t|d| |d<||d|<q |S)u 将512x512图像的绘图项坐标还原到原始图像尺寸 参考主程序detect_only的坐标还原逻辑 Args: draw_items: 绘图项列表（512x512尺寸） scale_x: X方向缩放比例 scale_y: Y方向缩放比例 Returns: dict: 还原到原始图像尺寸的绘图项字典 type rectangletopLeft bottomRightrrrrtextpositionpolygonpointscirclecenterradiusritem_) enumeratecopyr#floatappend) r7rrrriitemZ scaled_itemZ scaled_pointsr<Z avg_scaler-r-r.rsP        z-EarFoldDetector._scale_draw_items_to_originalc Csi}i|d<i|d<g|dd<g|dd<d|d<|di}|dd }|d d }|d d }|dd}|di} | dd} |dd} |di} |di} | dd }| dd}| dd}| dd}| dd}| d d!}| d"d#}| d$d%}| d&d}| d'd(}| d)d*}| d+d}| d,d-}| d.d-}| d/d}| d0d(}| d1d}| d2d}| d3d4} | d5d6}!| d7d8}"| d9d}#| d:d }$| d;d<}%|d=i}&|&dd }'|&dd>}(|d?i})|)d@d}*|)dAd}+|},|,jdBdC\}-}.t|,||f},tdD|.dE|-dF|dE|t}/|/}0|j||j }1|1sWtdG|jj |dd dH|jj |SdIdJdKdLdMdNdO}2|j |2| |j_ ||j_| r| dPdQ| dRdQ| dSdQ| dTdQ| dUdQ| dVdQdW}3|3D] \}4}5|j|4|5q|jNz|jj|,|j ||ddX}6Wn7ty}7z*tdYt|7|dd dZt|7d|d[<|WYdB}7~7WdBSdB}7~7wwWdBn 1swYt}8|8|0}9td\|8|0d]d^|j}:|:|6};t}|;D]}?|?da|2vrC|2|?da|?da<q2t}@tdb|@|<d]d^d}At|;dckr_d}A|r%|d-ksl|d-kr|,jdBdC\}-}.t|.|}Bt|-|}Ctdd|.dE|-dF|BdE|Ctde|df|tj|,|B|Cftjdg},|;D]f}?|?dhi}D|Dr t|Ddi||Ddi<t|Ddj||Ddj<t|Ddk||Ddk<t|Ddl||Ddl<t|Ddm||Ddm<t|Ddn||Ddn<|Ddk|Ddi|Ddo<|Ddl|Ddj|Ddp<|Ddo|Ddp|?dq<qtdrt|;dsn tdt|du|dvt}Etdw|E|@d]d^g}Fg}Gd}Hd}Id}Jd}Kd}Ld8}Md}N|;D];}?|?dadIkri}O|?dhdi|OdE<|?dhdj|Odx<|?dhdo|Odo<|?dhdp|Odp<t|,|O}P|P|?dy<|F |OqHd}Qt|Fdckr|rd}Q|;D]}?i}O|?dhdi|OdE<|?dhdj|Odx<|?dhdo|Odo<|?dhdp|Odp<t|,|O}P|P|?dy<d}R|?dadKkrd}H|?dadJkrd}I|?dadNkrd}J|?dadLkrd}K|?dadMkrd}L|Qrt|Fd8krq|Fd8}Sdzd{}T|?dadKkr)|T|O|S}U|U|kr)d|?d|<d}Rtd}|Ud~d|d|?dadJkrL|T|O|S}U|U|krLd|?d|<d}Rtd|Ud~d|d|?dadNkro|T|O|S}U|U|krod|?d|<d}Rtd|Ud~d|d|?dadLkr|OdE|OdodC}V|SdE}W|V|Wkrd|?d|<d}Rtd|Vd|Wd|?dadMkr|OdE|OdodC}V|SdE}W|V|Wkrd|?d|<d}Rtd|Vd|Wd|?dadJkrd|?da<|?dadIkrِqi}X|P|Xdy<|O|Xd<|?da|Xda<|?d|Xd<| rZ|?dadvrZt|FdckrZt| |Odp|Odo| dkrP|OdE|Fd8dE|Fd8do| dkrP|Odp|Fd8dp| dkrPd|?d|<d}Ntd|Odp|Odod|?ddq|RrX|G |Xq|Rrq|G |Xtd|?dd|?dadqt}Ytd|Y|Ed]d^d }ZdB}Sd}[t|Fd8krt|Gd8kr|dd ddc}Mn |dd ddc}Mt|Fdckr|dd d|Zd}Zdc}Md}\d}]t|Fdckr|Fd8}S|Sdo||kr|dd ddc}M|Zd}Z|Sdp||kr|dd ddc}M|Zd}Zt|Gd8kr |GD]}^|^d}_|^dy}`|j|S|_}a|a|_d<d8|^d<|_dE|SdEdCkrS|add|Sdod*krQdc|^d<dc}M|dd d|Zd}Zq |^dadKkrtt|_dE|SdEt|Sdp|]}]|Md8krtdC}M|r|_do|Sdo|kr|_dp|_do|krdc}M|dd d|dd dq |^dadNkr tt|_dE|SdE|Sdot|Sdo|\}\t|_dE|SdE|Sdo}b|b|Sdo|krt|Gdckr|Md8ks|MdCkrd}Mndc}M|Zd}Zn |Md8ks|MdCkrd}M|dd dq |_dE|_dodC}c|_dx|_dpdC}d|Sdx|SdpdC}e|_dE|_dxf|_dE|_do|_dxf|_dE|_do|_dx|_dpf|_dE|_dx|_dpfg}f|d|ekrj|fd8}g|fdC}h|fdc}id}jn|fdc}g|fd}h|fdC}id}jt |i|g|h}k|c|df|j|i|k|g|hgd|^d<d}l|kd8|SdE|Sdo|krdc}M|Zd}Z|dd dd}l|_dE|SdE|Sdo|kr|lsdc}M|Zd}Z|dd dtd|d|_dE|SdEd|Sdo|dd}l|ls|Mdckrd}M|dd dq |\d-kr|]dQkrdc}M|Zd}Zt}mtd|m|Yd]d^d}n|j!"|'}o|*r4|Mdckr4|Md8kr4d}n|+r9d}n|Nr>d}ni}pdB}q|SdBur|S#}q|qdEt||qdE<|qdxt|qdpd*|qdx<|qdpdC|qdp<t||qdo<|qdEd8kr|qdot|qdE|qdo<d8|qdE<|q#}r|rdodC|rdo<t$|,|q}st|sd}sdB}t| r|qdBu rŐztd|SdBud| |SdBu ru| d8k ru|SdE|Sdx|Sdo|Sdpf\}u}v}w}xt|w| }yt|x| }z|u|w|ydC}{|v|x|zdC}|g}}t%dD]P}~t%dD]H}|||z|~dcd}|{|y|dcd}||,jd8k rC||,jdck rCt|,jdk r8|} t|,||d8fq|} t|,||fqqg}|sjdBdC\}}t%dD]F}~t%dD]>}||~dcd}||dcd}||k r||k rt|sjdk r| t|s||d8f q\| t|s||f q\ qVt|}dk r|}&t|}dCk r|}dcdn|}}t'|t|}|t|dC}|}d}nd8}}}t|dk r|&t|dCk r|dcdn|}t'|t|}|t|dC}|d}nd8}}}t||}t||}||||||||t||d }ttd|dd|dd|d|"|k o=|!kn}|"|k oJ|!kn}| rs| rV| ss|Mdck rrdc}M|Zd}Z|dd d|dÛd|dÛn$ n"td|SdBud| |SdBu rtdƃ| d8k rtd| dȝWn3t y}7ztdt|7|dd dt|7dB}tWYdB}7~7n dB}7~7wwtd||SdBu r>|n r>|o r/ddddddќ}|j! ||j!j(|sdddҍ}|j!)|}td|| r.|da|pd<|d|pd<|dadk s|dadk s|dadk r$d|pd<nd|pd<dc}M|Zd}Zn|j!j |pd<|o s>|j!j |pd<t}||m|pd<td||md]d^d|d<|Md8k rad|d<nO|Mdck rkd|d<nE|MdCk ryd|d<d|d<n7|Mdk rd|d<d|d<n)|Mdk rd|d<d|d<n|Mdk rd|d<d|d<n |Mdk rd|d<d|d<g}g}|;D]}?d||?v r q|?dhi}D|D r\|Ddi|Ddj}}|Ddk|Ddl}}|?dad}4|?dd8}|4d|d~}| |d8d6d8dd6d8d8dd6d6d8dd6d8d6dd8d6d6dd6dd8ddW}||4dddd}dt|t|dt|t|dd8dC|dd}| ||4d|d~dv}d|dd|t|t|dddd}| | q|GD] }^d||^v rj q`d|^v rj|^d}|d}|d}|d}|\}g}hdddcdCd6d8d8dt|gd8t|gdcdt|hd8t|hdcdgdd}| |ddd8dCd8d6d8dt|d8t|dcdt|d8t|dcdgdd}| ||D].}ddd8dcd6d6d8dt|d8t|dcdt|d8t|dcdgdd}| | qdt|d8t|dcddd8dCd8d8d6ddd}| |dt|d8t|dcddd8dCd8d6d8ddd}| ||D]!}dt|d8t|dcddd8dcd6d8d8ddd}| | qH q`|SdBu rt|SdE|Sdo|}ddd8dd6d8d8dt|t|Sdxdt|t|Sdx|Sdpdgdd}| |t|SdE|Sdo|}dt|SdEt|Sdxdt|t|Sdx|SdpddcdCd6d8d8ddd}| |d|pvrbd|pdd|pdd|pdd]d}|pddkrd8d6d8dnd6d8d8d}d|dd|ddddd}| |dt*vrb|pddkr/d8d6d8dnd6d8d8d}dt|qdEt|qdxdt|qdE|qdot|qdx|qdpdd8dC|dd}| |d|9d]dd|dd|j+dg}t,|D]E\}~}|Md8kr|~dckrd8d6d8dnd8d6d8d}|ddkr|~dckrd6d8d8d}d|dd|dd|~dddd}| |q~|tdBurtd|tddd|tddd |tddd |td d}d |tddd |td d}d|ddd6d6d8dddddd}| || |d|tddd |tdd}d|tddd |tdd}d|ddd6d6d8dddddd}| || |d|tddd|tdd}d|tddd|tdd}d|ddd8d6d6dddddd}| || |ntdtdd|Zddd8d6d6dddddd}| || |Z||d<||d<|S(uR 核心检测逻辑 - 基于原始run_delect方法的完整实现 result_imager 算法错误 算法消息rrrcrerhrlrmrnrorfrgrxryF横向纹理附加检测T横向纹理附加检测参数极耳翻折用户参数*翻折保护距离（极耳高度比例）0翻折起点保护距离（极耳高度比例）{Gz?0翻折起点扩展距离（极耳高度比例）极耳高度（像素）极耳高度保护阈值333333?极耳宽度（像素）极耳宽度保护阈值?分层保护阈值皙?月牙宽高NG 月牙宽度皙? 月牙高度?启用自动缩放功能检测后图片高度缩放?检测后图片宽度缩放启用相对位置筛选器重叠度阈值启用灰度检测灰度范围NG 灰度内缩ffffff?灰度差值MAXri灰度差值MINr 算法存图算法存图路径"算法存图最大占用硬盘(MB)rdr隔膜用户参数启用隔膜分析 隔膜全检Nru🔧 缩放图像: rz -> 模型加载失败: u翻折模型加载失败: rprqrrrsrtrurvrzr{r|r}r~rrrru❌ 模型推理失败: u翻折模型推理失败: ru翻折模型推理时间: .4f 秒u提取检测列表时间: g333333? 类型字符u合并检测列表时间: ru🔧 自动缩放图像: u 宽度缩放比例: u, 高度缩放比例: ) interpolation AOI位置r?r@r(r)center_xcenter_yrr面积u5✅ 图像和检测结果缩放完成，共缩放了 u 个检测框u*🔧 未启用图像缩放 (宽度比例: u, 高度比例: )u自动缩放时间: rr$cSst|d|d}t|d|d}t|d|d|d|d}t|d|d|d|d}||ks<||kr>dS||||}|d|d}|dkrTdS||S)u-计算item_aoi被container_aoi包含的比例rrrrrr)r r!)Zitem_aoiZ container_aoir?r@r(r)Zintersection_areaZ item_arear-r-r.calculate_containment_ratios""zCEarFoldDetector._run_detection..calculate_containment_ratiou 不显示u 月牙检测项包含度不足: .3fz < u，设置为不显示u 折痕检测项包含度不足: u 分层检测项包含度不足: u&内插未超出左边界: item_center=z > je_left=u&内嵌未超出左边界: item_center=zhehengr%索引)r横纹最低长宽比横纹最低风险高度横纹最低风险宽度u5🔍 检测到可能的横向纹理误检，长宽比u ，AOI索引 u 被标记为不显示u AOI索引 u ，类型 u4 被添加到其他AOI列表，包含度检查通过u检测框处理时间: u ❌ 未检测到极耳,有翻折u%❌ 未检测到极耳及其他翻折u;❌ 检测到多个极耳,理论上拆图只拆单个极耳zduojier; u1❌ 极耳高度过低，判定为首极耳翻折z gaoduguodi;u(❌ 极耳宽度过大，判定为错位zcuowei;u与极耳关系posu 位置关系u 水平距离u❌ 检测到极耳翻折zneicha;u❌ 检测到月牙宽高异常u 检测到月牙z dafenceng;u 检测到分层u7上半部分，沿左上-右下对角线镜像右上角ru7下半部分，沿右上-左下对角线镜像右下角)u 中心位置u 位置描述 原始角点 镜像点对角线端点u 镜像分析z dafanzhe;zfanzheguoshen;u! 保护距离判定:判定阈值u 水平距离 u > 保护距离 u&，且未被之前的翻折判定为NGu 检测到小翻折z dafencengC2;u翻折识别时间: )rru&🔍 灰度检测条件检查: je_aoi=z, gray_inward_ratio=) je_mean je_medianje_maxgm_mean gm_mediangm_maxmax_diff mean_diff median_diffu*✅ 灰度检测数据已创建: JE均值=z.1fu , GM均值=u , 差值=zhuiduNG;u(❌ 灰度差值异常: 最高值差值=ru, 均值差值=u"⚠️ 跳过灰度检测: je_aoi=u2 - je_aoi为空，检查极耳检测是否成功z - gray_inward_ratio=u ，需要>0u❌ 灰度检测异常: u灰度检测异常: u1ℹ️ 灰度检测已禁用: gray_detect_enable=OKZfanguangZfanguang_pluseZdazhouZneicha)rrrrr )Ztop_kZuse_yolo_preprocessu隔膜分类Top1结果: u类别u分类置信度 置信度zgemoNG;u 模型错误u 推理耗时u隔膜识别时间: u月牙u异常u 无极耳u 多极耳u 小翻折u分层或月牙unknown综合置信度:)rgrBrr)rrr lineStylepenWidthcolorshow(rZArial )rrZ fontFamilyZfontSizer rr!rrrr)rZisClosedrrr rr!rg@)rrrrrr r!g@zGM: ,zs)g$@gY@gm_aoizModel: szResult: zcount: dg9@u(🔍 准备显示灰度检测数据: JE=r z, GM=ru极耳: 平均值=u 中值=r z JE: Mean=z Median=g@_@u隔膜: 平均值=rz GM: Mean=gb@u极耳，隔膜: 平均值差=ru 中值差=rzDiff: Mean Diff=z Median Diff=ge@u+⚠️ 灰度检测数据为空，未显示u准备显示结果数据gi@rr)-r#rcv2resizerrrrMr;rTr5rrrrrrrRrr"r`extract_detection_list$merge_overlapping_detections_inplacelenrZ INTER_LINEARr r/Zcalculate_aoi_relationshipsr!absrrErNrrget_cv_img_fronm_aoi_exrangesortsumrZget_top1_resultlocalsrUr)r7r$rr*r model_pathZmodel_confidence_thresholdZmodel_iou_thresholdrjrrZhorizontal_texture_checkhorizontal_texture_paramZ usr_paramsZfanzhe_protect_ratioZfanzhe_protect_startZfanzhe_protect_expandZ JE_heightZJE_height_ratioZJE_widthZJE_width_ratioZJE_fenceng_thresholdZyueya_wh_ng_enableZyueya_width_ngZyueya_height_ngZenable_auto_scalingZdetect_img_height_scaleZdetect_img_width_scaleZenable_pos_filterZpos_filter_thresholdZgray_detect_enableZgray_ng_enableZgray_inward_ratioZ gray_diff_maxZ gray_diff_minZsave_image_by_modelZsave_image_pathZsave_image_max_sizerZgm_cls_model_pathZgm_cls_model_imgszZgm_cls_usr_paramsZ enable_gm_clsZenable_gm_cls_all src_imagerrrZ model_start load_resultrrrr model_resultsr,Z model_endZ model_costZcur_delect_modelZ deleted_listextract_detection_list_end merge_success merge_msgZ detect_itemZ merge_endZhas_other_itemZ new_widthZ new_heightZaoi_posZ auto_bg_endZ je_aoi_listZother_aoi_listZ has_yueyaZ has_fanzheZ has_fencengZ has_neichaZ has_neiqianZ fanzhe_resultZhas_horizontal_texturetemp_aoiZimage0Zuse_pos_filterZ need_appendZje_aoirZ containmentZ item_centerZje_left temp_itemZ aoi_seg_endZng_infoZsee_gmZ fenceng_upZ yueya_downZ other_itemZ other_aoiZ other_imageZother_aoi_resultZ fenceng_posZother_center_xZother_center_yZ je_center_yZ other_cornersZ diagonal_p1Z diagonal_p2Z target_pointZ position_descZcur_pntZisNGZ check_je_endZ use_gm_cnnZcls_load_successZ gm_resultr&Zjegm_aoiZ gm_image_clsZgray_detect_dataZje_xZje_yZje_wZje_hZinward_wZinward_hZinward_start_xZinward_start_yZ je_pointsrjrrZ gm_pointsZgm_hZgm_wZ je_filteredr r rZ gm_filteredrrrrrZmax_diff_in_rangeZmean_diff_in_rangeZcustom_class_namesresultsZ top1_resultZ gm_je_endrrr?r@r(r)Z confidenceZconf_strZ color_mapr Z rect_itemZ label_textZ text_itemZmirror_analysisZ mirror_pointZoriginal_pointZdiagonal_pointsZ diagonal_lineZ mirror_lineZ diag_pointZconnection_lineZoriginal_circleZ mirror_circleZ diag_circleZ protect_xZ protect_lineZstart_rect_rightZ protect_rectZgm_textZ gm_text_colorZ gm_text_itemZ gm_rect_colorZ gm_rect_itemZ status_textsrZ text_colorZtemp_strZ temp_str_enZ je_text_itemZdiff_text_itemZ ng_text_itemr-r-r.rs                                                                        "*&   & ,.    $0      $                  "      *                                                  && &( ,  ($$   $$   $$          zEarFoldDetector._run_detectioncCsddiS)u< 获取检测器可用的检测函数列表 ru翻折检测（仅检测）r-r6r-r-r.get_available_functionssz'EarFoldDetector.get_available_functions function_namecCs"|dkr |||Std|)u- 调用指定的检测函数 ru未知的函数名称: )r ValueError)r7rAr$rr-r-r. call_functions zEarFoldDetector.call_functionc Cs(|jsddiS|dur |n|j}z]t}i}i}i}|||}d|vr,|d|d<d|vr6|d|d<d|vr@|d|d<d|vrJ|d|d<d|vrYtd d |d<|WSg|d <d |d vr|d d } z(|| |}d |vr|d D] } |d d| qxd|vrd |d<|WWSWn'ty} ztdt | dt | |d<|WYd} ~ WSd} ~ wwd|d vr|d d} z(|| |}d |vr|d D] } |d d| qd|vrd |d<|WWSWn(ty} ztdt | dt | |d<|WYd} ~ WSd} ~ ww||d<||d<| ||||d<|r=| dd| dd| d d|rR| dd| dd| d dd |vr]| d dt} td| |dd|WSty} ztdt | ddt | iWYd} ~ Sd} ~ ww)uo 完整检测流程：极耳裁切 + 翻折检测 从主程序迁移的 run_all 函数 rrN上极耳裁切框下极耳裁切框 上极耳AOI 下极耳AOIru+❌ 检测异常结束，跳过后续步骤rr up_jr_imageru 上极耳:u❌ 上极耳检测失败: u上极耳检测失败: down_jr_imageu 下极耳:u❌ 下极耳检测失败: u下极耳检测失败: u上极耳结果u下极耳结果rrrrru❌ 完整检测异常: u完整检测异常: ) r_r\rr_run_cutrrrr"r`_restore_draw_items_to_originalpop)r7r$rrrr* result_up result_down result_allZup_imagerr,Z down_imagerr-r-r.run_alls                    zEarFoldDetector.run_allrhrrc Csddl}||krtd|ddSt|tr_|D]A\}}|r(|d|n|}t||jr=td|d|jdqt|ttfrO|||||d qtd |d t |j qdSt|trt |D]F\} } |rv|d | d nd | d }t| |jrtd|d| jdqht| ttfr|| |||d qhtd |d t | j qhdSt||jrtd|d|jddStd |d t |j dS)uA递归打印字典中每个值的数据类型，查找numpy数组rNz u: <递归深度超限>.u ❌ z: numpy.ndarray (shape=u) <-- 发现numpy数组!ru ✅ z: []) numpyr isinstancedictrndarrayrlist_debug_print_dict_typesrr0r) r7datar9 max_depthZ current_depthrkeyvalueZ current_pathrrr-r-r.rY]s2      z'EarFoldDetector._debug_print_dict_typescF Cs i}i|d<i|d<g|dd<g|dd<d|d<|di}|dd }|d d }|d d }|dd}|dd} |di} | dd} | dd} | dd} | dd}| dd}| dd}| dd}| dd}t}|}|jdd \}}t|||f}t}td!||d"d#|j||j }|std$|jj |dd d%|jj |St}td&||d"d#d'|j_ ||j_ d(d)i}|j||jJz|jj||j ||d'd*}Wn3ty#}z&td+||dd d+|d|d,<|WYd}~WdSd}~wwWdn 1s/wYt}td-||d"d#|j|}t} td.| |d"d#|j|d/\}!}"t}#td0|#| d"d#|D]}$||}%||}&t|$d1d2|%}'t|$d1d3|&}(t|$d1d4|%})t|$d1d5|&}*|'|)d }+|(|*d },|,|d k}-|-r| }.| }/n| }.|}/|+t|/d6|d }0|,t|.d6|d }1t|/d6|}2t|.d6|}3|0|/}0|2|/}2|0d7ks |1d7ks |0|2|ks |1|3|kr=td8|$d9d:td;|$d<d=td>|0d?|1d@|2dA|3tdB|dA|dC|$d<<qrt|dDdEddFdd }g}4|D]}$|$}5i}6||}%||}&t|$d1d2|%|6dG<t|$d1d3|&|6dH<t|$d1d4|%|6d4<t|$d1d5|&|6d5<|6|5dI<t|$d1d2|%|5d1d2<t|$d1d3|&|5d1d3<t|$d1dJ|%|5d1dJ<t|$d1dK|&|5d1dK<t|$d1dL|%|5d1dL<t|$d1dM|&|5d1dM<t|$d1d4|%|5d1d4<t|$d1d5|&|5d1d5<|5d1d4|5d1d5|5dN<|4 |5qPt}7tdO|7|#d"d#t|4d6kr|4d7}8|8d1dM}9|9|d k}-|8}:|8d1|:d1<|-rttdP||:d1dM|7<|:d1d3|7<|:d1dK|7<n%tdQ||:d1dM|8<|:d1d3|8<|:d1dK|8<|:d1d4|:d1d5|:dN<d6|:dR<|4 |:tdSt|4dTn"t|4d krtdUt|4dV|dd dWt|4dV|Sd};d}|=d1dM|>d1dMkr|=};|>}};|=}<|;d1|dX<|sz*EarFoldDetector._run_cut..)r\reverserrr%r(r)rrru还原检测框时间: u/检测到上极耳，生成下极耳，间距: u/检测到下极耳，生成上极耳，间距: ru/✅ 成功生成第二个极耳，现在共有 u 个极耳u'⚠️ 检测到的极耳数量异常: u ，预期为2u 检测到的极耳数量异常: rFrGZcut_aoirDrEu上极耳裁切框:u下极耳裁切框:)rgrgrHrI)rrrr)r?r@r(r)rrrr)u 原始尺寸u缩放后尺寸缩放比例_x缩放比例_y) 裁切区域u 检测框 缩放信息)rr) 上极耳 下极耳u源图像尺寸极耳裁切信息ru裁切出图执行时间: )r#rrrr)r*rrLr;rTr5rrrrrQrr"r+r,rsortedrr-r`r r/r )Fr7r$rr*Z cut_paramsrZcut_confidenceZcut_iou_thresholdZcut_input_sizeZcut_merge_detectZusr_dictZ top_jr_widthZ top_jr_heightZbottom_jr_widthZbottom_jr_heightZjr_width_expand_ratioZjr_height_expand_ratioZ jier_spacingZ gm_get_deeprr6Z src_heightZ src_widthZ resize_imageZ resize_timer7Z cut_load_timerr8r,Zcut_predict_timeZ cut_jr_listr9r:r;Zcut_merge_detect_endrjrrZorig_x1Zorig_y1Z orig_widthZ orig_heightZ orig_center_xZ orig_center_yZ is_up_jierZjr_width_paramZjr_height_paramZ cut_aoi_xZ cut_aoi_yZ cut_aoi_widthZcut_aoi_heightZ src_jr_listr=r<Zaoi_resize_timeZ single_jierZsingle_center_yZ second_jierZup_jierZ down_jierZjier1Zjier2rHrIZcut_timeZ up_actual_xZ up_actual_yZ down_actual_xZ down_actual_yr-r-r.rJ}s                                  $$$$                               1zEarFoldDetector._run_cutcCsdd}|di}|siSi}d}|rd|vr|dD]}|} d| d<|dd kr]||d d |d d d|\} } ||d d |d d d|\} } | | d| d <| | d| d <n|ddkr||dd |dd d|\}}||d| d<|ddd|dddd}|d|| d<nG|ddkrg}|dD]}||d |d d|\}}|||dq|| d<n|ddkr||dd |dd d|\}}||d| d<d|d}| ||<|d7}q|rd|vr|dD]}|} d| d<|dd kr:||d d |d d d|\} } ||d d |d d d|\} } | | d| d <| | d| d <n|ddkrt||dd |dd d|\}}||d| d<|ddd|dddd}|d|| d<nJ|ddkrg}|dD]}||d |d d|\}}|||dq|| d<n|ddkr||dd |dd d|\}}||d| d<d|d}| ||<|d7}q|S)u 将512x512裁切图像的检测结果复原到原图坐标系 从主程序的 restore_draw_items_to_original 方法迁移实现 Args: result_up: 上极耳检测结果 result_down: 下极耳检测结果 result_all: 裁切结果，包含极耳裁切信息 Returns: dict: 复原到原图的绘图项字典，每个项目以唯一ID为键 c SsX||}||dd}||dd}||dd}||dd}t|t|fS)u1将512x512裁切图像坐标复原到原图坐标rprmrnrorr)r) Zx_cropZy_cropZear_typecut_infoZear_infoZx_cutZy_cutZ x_originalZ y_originalr-r-r.restore_coordinates_to_originals zXEarFoldDetector._restore_draw_items_to_original..restore_coordinates_to_originalrsrrrqu来源rrrrrrrrrrprmrnrrrrrrrZ03drrr)r#rr)r7rMrNrOrvruZ result_dictZ item_counterrZ restored_itemr?r@r(r)ZcxZcyZ scale_avgZrestored_pointsr<r=r>ZtxZtyZitem_keyr-r-r.rKs                  z/EarFoldDetector._restore_draw_items_to_originalc Cs dddddd}ddddd d d d d }dd d d d d d d}dd d d d d d d}dd dd}dddd|d|d}idddd dd dddddddddd dddd d!d d"dd#d$d%d$d&dd'd d(dddd)d*dd+d,d-}dd.d/}ddd0}ddd1} ||||||d2} | S)3uN 获取默认参数配置（从原始配置文件中提取） rhrmrorgT)rerr^rfr_rrr{rrhr)rarbrcrdrerfrgriF)ryrzr|r}r~rrg @r)rrr)rerlrnrfrxrrrrrrrrrrrrrrrrrrrrrrrrirr9r)rrrrrrrr)rerf)rr)u更新脚本地址u更新UI地址)rbr`rcrrdrr-) r7Z jrcut_dictZjrcut_usr_dictrr5Z deleted_dictZ fold_usr_dictZgm_dictZ gm_usr_dictZ update_dictZall_parametersr-r-r.r]Xs           z&EarFoldDetector.get_default_parameterscCs&|jr td|dStddS)u 重写基类的预热方法 如果检测器已初始化，再次执行模型预热 如果未初始化，只记录警告信息 u🔥 执行检测器预热...u1⚠️ 检测器未初始化，无法执行预热N)r_rr^r6r-r-r.warm_ups  zEarFoldDetector.warm_upcs(td|_d|_d|_tddS)u 清理资源Nu🧹 检测器资源已清理)rKcleanuprLrMrNrr6rWr-r.rxs  zEarFoldDetector.cleanup)N)rhrr)rYN)r0r1r2__doc__r8r`rZrrboolrar^rrWrrrr@rCrPrYrJrKr]rwrx __classcell__r-r-rWr.rFs: !*u>(F(.* g( J } rF)%ryr)rTrrtypingrrrrrrOjsonossysrrr9dirname__file__Z current_dirZ parent_dirr base_detectorr Zutils.model_utilsr Zutils.model_cls_utilsr r ImportErrorr,rr r4rErFr-r-r-r.s>        %74