IjyhdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl Z e jdddl mZddlZddlmZddlZddlmZmZddlmcmZdd lmZmZmZdd lmZe j@jCd d d edz Z"ddD\Z#Z$Z%Z&dZ'dZ(ejRfd)dZ*d*dZ+d+dZ,d,dZ-d-dZ.d.dZ/d/dZ0GddejbZ2d0dZ3d1dZ4d2dZ5d3dZ6ejnd4d Z8d5d!Z9d6d"Z:d7d#Z;d8d$Zd;d'Z?e@d(k(re?yy)r sp.odsTz.osprocessed_inputsmodelsoutputsfeature_attributions>@)image_2dvoxel_3d point_6000 graph_6000c0tj||S)Ndtype)torch as_tensor)xr!s rtensorr%%s ??1E **c|djtjj}|djt}|djt }t |\}}|j}tjt|t}d|d|dfd|d|d ffD]Z\}}} td d D]E} || d } | |vs| | d z s|| d z ||j| <d ||j| <G\||z} ttj| |ddtj||d} tj| |jtj|jtjf||Dcgc]}|d c}z|Dcgc]}|d c}zfScc}wcc}w)N process_xprocess_feature_validprocess_feature_namesu 상부온도 heater_top_cheater_top_imputedu 하부온도heater_bottom_cheater_bottom_imputed02dTrz :observedz:imputed)astypenpfloat32copyboolstrr tolistzeroslenrangeindexr whereasarrayr_)itemrawobservednamesbaseunitsimputedprefixvaluesflagszonervalidvaluens rprocess_vectorrN)s. {  " "2:: . 3 3 5C$G^B_BfBfgkBlx ( ) 0 0 5EEYZ_E`{tUimititivbfhhs3x&G#143GNbIc"dguw{}NxOQUVmQngo"p!RLDXd3Z(Dt|dQh(.tax(8DJJt$%VZ'$**TBR:S!"p w E $RXXeS!%E 7755=#r * 1 1"** ==r&c t|j|t|dzt|d\}}t j |}|ddddf}|j d|kr^tj|t jt jt|dddf||j dz df}|jtjS)z?Deterministic local message neighborhoods; first query is self.r/)kworkersNrT) rqueryminr:r3 atleast_2dshapec_repeataranger2int64) coordinatescount_r<results r knn_neighborsrn>s{#))+UQYKHX9Ycd)eHAu MM% E 1ab5\F ||Avryy3{3C)DQW)MuW]WcWcdeWfOfmnoop == ""r&c j |dk(rtjtdz d5}|d|dzj}tj|}|tj dt |dz td t |tj }|d jd d |}|djd d |}|djddd djd |djd|}|djddd djd |djd|}dddtjdzdz|dzdz|dzf}t|t|dddt |ddfS|dk(rtjtdz d5}|djtj } |d| t#|dzz}|d|d}}dddtj$|t|t dddt |d dfS|d!k(rtjtd"z d5}tj dt |d#dz d tj }|d#||d||d||d|f\}}}}dddt|t|d$d"d%t |d&dfS|d'k(rtjtd(z d5}|d#|d|d|d|d)f\}}}}} tj&|} tj(t |dftj } tj*j-| | d|| dtj*j-| dddf| ddtj|| tj.| dz f}tj||f}t1t |D cgc]} g}} | j2D]+\}}|t5|j7t5|-tj8t;|Dcgc]\}}|dd$|gd$zzdd$c}}tj}dddt|d(d*t |d+d,t |d-dfSt=|#1swYxYw#1swYxYw#1swYRxYwcc} wcc}}w#1swYtxYw).Nr+data_processed/images/cad_multiview_128.npzF allow_pickle silhouette visibilityrr/ipr xyz_mmnormalr$ feature_validi@zOsix CAD orthographic views with explicit 2-D pixel-neighbor message aggregationz@4 nearest selected image-grid pixels in the same view; not a CNN)sourcerepresentationnodesdenoiser_adapterrz/data_processed/voxels/cad_surface_sparse_64.npz voxel_coordsgrid_origin_mmvoxel_pitch_mmzM64-grid sparse CAD surface voxels with 3-D voxel-neighbor message aggregationz;6 nearest sparse voxel-grid neighbours; not a dense 3-D CNNrz,data_processed/points/cad_surface_points.npzpos_mmzOdeterministic 6,000 CAD surface points with PointNet-style local/global poolingz<8-nearest local pooling plus per-run global max/mean contextrz$data_processed/graphs/graph_6000.npz edge_indexzCAD graph nodes (,z< nodes) with edge-index message passing at each denoise stepzAprepared graph edge-index, up to 8 adjacent CAD nodes per message)r3loadr ravel flatnonzerolinspacer:rcrireshape transposererfr^rnr2r4float empty_like zeros_liker9addatrXr;Tintappendr> enumerate ValueError)rakeepidsposrxr$r\image_coordinatescoordsedgesummeddegreerl adjacencyr~ destinationr<row neighbourss rload_representationrHs z WWTIIX] ^bclOa o5<<>DbnnUYFZbkk!SX\3tSX3FbhhWXCH+%%b!,S1CAhKOTW=WY\_bYb"bcFK40$ FWYZ8[iVj{FIJMFNdf^gg g z WWTMM\a bfg~&--bjj9FCSATW]`efghxfy`zWzAz3fa0tAcBMM#& At(t|dz fi|t|dz jAtCjDi|tG|djHdd|ddd d!"d#$|jKDcic]\}}|| c}}d%|izScc} }wcc}} wcc}}wcc}}w)&zMMap each prepared representation onto the paired CAD target only by geometry.Tparentsexist_okrr/ cad_pos_mm cad_normal)ra)rUkeepdimsgư>heater_projection_valid%provisional_heater_top_c_per_cad_node(provisional_heater_bottom_c_per_cad_node'provisional_heater_delta_c_per_cad_nodeg@@g@@NrTcad_to_scan_normal_residual_mmweak_defect_probabilityweak_defect_mask defect_class)rrrxnearest_paired_cad_node conditionsigned_target_scaled risk_target risk_binary nodes_per_runneighbor_index_one_runr*rrepresentation_featuresrepresentation_feature_validrun_defect_class scan_points__cad_mmscan_pos_cad_mmz.npzz.jsonrzcKDTree CAD position onlyzweak/unvalidated diagnostic) condition_dimnearest_paired_cad_mappingr label_statusFry ensure_asciiindentutf-8encodinginfo)&INPmkdirrrrrbr3isfiniteallanyrXlinalgnormrNr6r@rffullr:rr4rgr2rr[SCALE_MMr>int32 concatenater7updatesavez_compressed write_textjsondumpsrreitems)rrrxfeaturesr\rrrcad_posrrlnearest all_condition all_targetall_risk all_binary all_local process_namesrunr@procrKlocal local_maskrroutputrrLs rprepare_variantrxsrIIdTI*4G4M1C4TYgYiwajmL)G 1 l8S:!''Q'7JAw ;;v  " " $BFF6NZPWEXF bjjQFMMFMjbXR*2jl} T,T2mT3499;<dBCGLdS}N~@GOHJNOxJyzAJBBCFJJWWc#h]E%L"**E uPZGZuEE#+vuj"))DQUJX[\_X`ghBiksuyvAvABDBLBLvMMN Y--bjj9:J%%djj2~D2~3vVaObOhOhijOkKlMhEN`}2~MR[\']ho%p)/ 8:3CJ 8FD> II $:[ S ` 9s' S &S&S < S$c.eZdZdZddfd ZddZxZS)TopologyConditionalDDPMa)Denoiser with a representation-specific local message every DDPM step. ``point_6000`` additionally follows the PointNet pattern: local kNN pooling plus global mean/max pooling for each process run. The other three variants use their own 2-D, 3-D or CAD-graph neighbourhood index. c,t |||_||_t j t |Dcgc] }|||zz c}d}|jdtj|tjtjtj||tjtj||tj|_tjtjd|tjtj|||_||dk(rdndz} tjtjd|z| z|z|tjtj|d|_tjtj|| z|d ztjtj|d zd|_ycc}w) NrrTrr rrwr/ry)super__init__rvariantr3rr;register_bufferr"r#longr SequentialLinearSiLUcter) selfrrrrwidth run_countir message_width __class__s rrz TopologyConditionalDDPM.__init__sF }D.7dl~~U9EUVEUzAmO;EUV]^_ \5??4uzz+RS}}RYY}U;BGGIbiiPUV[F\]_]d]d]fgprp}p}AHHIKLQRSUSZSZS\]_]f]fglmr]sqtimiog&=1E }}RYYq5y='@5'HOPRPWPWPYZ\ZcZcdijkZlmvxwDwDEGENENOTWdOdejlmemEnoqovovoxy{yByBCHJKCKLMyNwOosou WsHc:|j|}||jjd}|jdk(r|j d|j |j d}|jddj|j|}|jddjj|j|}tj|||gd} n|} tjtjdtj dd|j" } |dddfj%|j"| dzd z } |j'tj| j)| j+gd} |j-tj|dddf|| | gdj/d|j1tj|| gdj/dfS) Nr/rrvT)keepdimrr{rr r)rrmeanrrrre expand_as reshape_asmaxrHr"catexprmathlogr!torsincosrsqueezer) rr$rrzrgroupsrrXmessagefphs rforwardzTopologyConditionalDDPM.forwards0 &&)1T__-2215U <<< '99R 2 21772;?FfkkRS\`kFaFkFklrFsF~F~@AGBLRLVLVWXaeLVLfLmLmLwLwx~LLJLJKLLMDKW\W`W`bghlmtauvwWxOVg ))ENN1dhhtnaqwwG HAaPTfILLYZY`Y`LabcdhbiLijmLm!quqwqwx}yByBDEDIDIDKLMLQLQLSCTUVyWrXopaeagaghmhqhqstuvw{u{s|}~FGHrIJKhLaMaUaUVWaXZ^Z`Z`afajajlmnukvwxayZzZBZBCDZEaEZEr&)`ry)rrrrr np.ndarrayrr7r rr rreturnNone)r$rrrrrr%ztuple[Tensor, Tensor])__name__ __module__ __qualname____doc__rr" __classcell__)r s@rrrs OEr&rc^tjdtjdd|z dS)Nr/g-C6?{Gz?r)r"cumprodr)stepss rscheduler0s&5==5>>$sSX;Y9YZ[+\$\r&c J|jsytj| d}||jt}t tj |tj dt|dzz |zj|jz S)Nstable)kindr/) rr3argsortr2rcumsumrhr:sum)scoreyorders raverage_precisionr:s 5574 **eV *Eahooe.DAUTVT]T]^_T`acajajklmpqrmstumuavTvwxTxS}S}SABAFAFAHTHNIGIr&cd|dzS)zDPortable raster label; exact Korean source names remain in NPZ/JSON.zprocess feature r/)r<s rascii_process_labelr=s eai[ ))r&c Ptjdt|dz tdt|t}tjdt|dz tdt|t}t j t||dt||dj}t|jdjj|jdjjzjdz S)Nrr/ir ry) r3rr:rcrr"cdistr%rrrHrr@)rbixiyds rchamferrDs {{1SVAXc#c!foC8BR[[3q6RS8TWX[\_`a\bTcjm=n"rwr}r}EFGHJFKLMQRSYZ[\^Z_S`aeSfsgsososqpqzABAFAFGHAIAPAPAUAUAWXYX]X]^_X`XgXgXlXlXnAn@t@t@vwx@xzysyr&c tjt|tjj |}t j t jdt|dz |tddd}t|D]\}}||tjt|ft|tj|\} } |d||z j| zz ||jz } |t|dz k(rdnt||dz} | dkr| n/|| j| zd|| z j| zz}|jddS)N) generatorrr/r rv)r"randnr: Generator manual_seedr3uniquerrrrrsqrtclamp) modelrabar sample_stepsseedr$seqr r`epsrlx0previouss rsamplerUs* kk#i.5??+<+H+H+NOAUWU^U^_a_j_jklmpqumvwxmxzFMP`QVRSWUWSWVXQT~!Aejj#a&3q6 CINATUWXY]^_Y`W`VfVfVhilVlTlnrstnunznzn|S|PRNOQTUXQYZ[Q[N[HJadehijklilemanGx@ABxBrtHLMUHVH[H[H]^`H`bcdhiqdrbraxaxaz{~a~H~pq 772a=r&ct|d}t|d}t|d} |djt} t|d} t |d} t |} t |jd| |d|| }tjj|j|d }t | | jz | jjdz jd }g}tdddf}t!|D]}|j#tj$d|t |f}tj&|}| |j)|zd| |z j)|zz}||||\}}t+j,||}t+j.|| |}|d|zz}|j1|j3tj4|j7j9}|j;|dzt|j7t|j7t|j7t=|| xsddt|j7}||dks||dz|j?jADcic]$\}}||j7jC&c}}f}|jE|d|jGtjHt!|D cgc]} tK||| |d| zc} j9}!tjL5|tjN|tjPt |tjR|\}"}dddtj4j9}|d|d}$}#g}%tU|dD]\}&}'tW|&| z|&dz| z}(|!dd|(fjYdtZz})|!dd|(fj]dtZz}*|j9|(tZz}+|#|$|)dddfzz},|#|$|+dddfzz}-| |(}.||(dk\}/t|/|.zj}0t=||(|.|0t_dt|/|.zjz d|0zt_dt|/j|.jzz d}1|d|'d}2|%j;ta|'ttc|)|+z jYttej(tejX|)|+z dztg|,|-tg|,|2|1t|*jYt|*j_dthjkddtjl|j?|jd| d ||d!d"d#th|d$z tejntp|d%z |d|#|$|!js|| | tZz|#d&|$d&|!js|| | dztZzz|js| | |d'tejtd() ||d!| t|jd||d|d|%|d*jwtath|d$z jytzd+ Scc}}wcc} w#1swYxYw),Nrrrrrrr/rr gh㈵>)lr weight_decay)rinfr) pos_weightffffff?rP)epochtotaldenoiseriskAPryi3(5r rrx?)rbz IoU_at_0.5z Dice_at_0.5rr)rMAE_mmRMSE_mmCAD_correspondent_Chamfer_mmregistered_scan_Chamfer_mmramean_stochastic_std_mmmax_stochastic_std_mmTrFrr) state_dictrrproduction_validrr.pt_generated.npz)NNr,constant_diagnostic_unlearnable_single_class)rrrgenerated_signed_normal_mmgenerated_pointcloud_mmrisk_probability defect_typedefect_type_statusr) rrrrhistory best_loss best_epochmetricsr checkpoint)>r%r2r6rr:r0rrer"optimAdamW parametersr6 clamp_minrLrr; zero_gradrandint randn_likerKFmse_loss binary_cross_entropy_with_logitsbackwardstepsigmoiddetachnumpyrr:rjrcloneload_state_dictevalstackrUno_gradrr9rrslicerrstdrr7rZr3rDMODrsaverOUTrr>r8 relative_tor )3rdataepochsrXdiffusion_stepssamplesrOrtargetrabinaryrMrunsrNrMoptpwhistbestr^rnoisextrRlogitr`rlosslossr7 loss_valuer`vr  generatedrlrrxrmjrslrrtgtpredrefr8r both risk_metricsscans3 r train_variantrs T+&'ItJRWRbRbciRjEostuovo{o{o}EpEGHIMNOIPGPFVFVFXY^F^p^lnjoprstu~j`i`cdiIJISISTWX]I^AHfgfHfHINOS_afb`eipqtuzqzizdh|@|I|I|KLOLTLTLV^c^k^klq^r^y^y^{^A^A^CX]EIEPEPZ_`aZajoptp{p{p}j~INOVO]O]O_I`hmnsnznzn|h}CTUZ[aCbChfhQiEj4;;=) Q z%'[`[k[k[m[s[s[u:v[uTWTUVW1QXXZ=M=M=O;O[u:v&w $q'"EJJLEKKHMNUHVIWHVCDPUV_`deqrz{|r|I}HVIW=X=^=^=`) 5)9)9&)A%++cRXk`e`j`jBklu#v(!U --  $ $ &E4>$x.FY[RX4>*# 1ac1W IadO$8$8$;H$Dt)TUVXTX/J]J]^_J`aiJicouo{o{o}ApBCKpKknRUV\]abcdhbh]iViRiMQorsyz}~@D~DzEsEoEknIOPRISGHW\]_W`bdWdUVknpqrspsoxoxozk{fjPabghjbklmPn|@ADEFGJLMNOLOKTKTKVGWAX|Xghimgmnqrstwxyx}x}x@A@E@E@GxGtHnIgIJJ}IQUXdehdiipVqQrLPtztAtAILMPIQ[`adeijmemanasasau[vAFGIGNGNOQOVOVX\]`X`cdWdOeGfAgG N O S T W G X v } ~ B C G v H P \ v { |  | D | D | F v G ` e f i f m f m f o ` p Bq tr +IIdDI)5::EDTDTDVgpgvgvwxgyKL`eptHLMSHTUgHh7ijmqupvvynzjz,{tfN334>UXdjGPGXGXY`aefgGhiqGqKNOXKYZ`ajZklul}l}~EFJKLMNlOZOPXZXKXkpkxkxy}~k@MQRdMey{yCyCDrystDL[^_h_n_nop_q[r~BOSTUOVdhijdkv|W[\uWvW}W}WMPRUY]X^^aVbRbQoQoptQuMv ww;wIW s%)]0 ]6A ];;^c t|d}t|jdt|d|d|t |d}|j t j|dd |j|d }d }|jjd}t jt |}t jt |t j } ||| |\} } | jj|jd d ||t |zfj!jdj#j%} |jjd}d} t j&t |f| t j }||||\}} t)d| }d|z j+ |z|j+z }|j!jj|jd d ||t |zfj!jdj#j%}t-j.dddd\}}|d| ddf|d|ddffD]\}}}}t1j2|dd d d d}|Dcgc]}t5t|c}d d d}|j7|||d d d|d|j9d|j;d|j=||j?dd |jAd!d"#|jC|d$|jEtF|d%z d&'t-jH|t1j2| d d d}t1j2|d d d}|d(d)d)tK|jMtNtQ|Dcgc]3\}}|dzt|dztK||tS| |d*5c}}tQ|Dcgc]3\}}|dzt|dztK||tS||d*5c}}d+}tF|d,z jUtWjX|d)d-d./y cc}wcc}}wcc}}w)0z?Fixed-state risk and denoised-deformation gradient diagnostics.rr/rrrrWT) weights_onlyrjr*rVr Nrry)rg@figsizeconstrained_layoutzweak-risk logit sensitivityz#495f9dz denoised deformation sensitivityz#a85b35irvzmean absolute gradient)colorlabelzmean absolute local gradientzXLS process feature indexz lower rightr)locfontsizer$g)\(?)rUalphazC: fixed-state local sensitivity diagnostic (not causal attribution)z_process_gradient.pngdpizabsolute local gradients with fixed diffusion state; risk uses mean logit at t=0, deformation uses mean |x0 estimate| at t=15 of 30 with x_t=0F)rankfeature_index_one_based source_namemean_absolute_gradient)rmethodrkcausalrx risk_featuresdeformation_featuresz_process_gradient.jsonrrr)-r%rrerr:rr"rrrrequires_grad_r9rrrgradrZrrrr0rKpltsubplotsr3r4r=barh set_xlabel set_ylabel set_titlelegendgridsuptitlesavefigATTRcloser7rr rrrrr) rrrxbase_conditionrMrCstartrzerot0rlr risk_valuertimestepepsilonrNrSdeformation_valuefigaxesaxrLtitlertopr<labels risk_orderdeformation_orderrpayloads r attributionrsG${+,N4KNL`L`abLcdghlm|h}d~@DE]@^_cnqrvwr@nA5BeDIDYDYZ_ZdZdeo}AZBCOZPDQSXS]S]S_ & 'Er""$33D9I CPYN@[4`e`k`klopylzBGBLBLaM]_W\]abdenWoOVOPQVpupzpzp|pEpEpGS\SaSabcdijopstypzjzdzbzS{SSSASFSFGHSISPSPSRSXSXSZHR""$33D9I4UZZY\]fYgXijnuzuuNAHLQRVW_`iLjBKBIJKpxy{p|}ApBkoHIJNHNGTGTGVFVW^F^_c_h_h_jFjCEkmkqkqkskxkxkzkCkCkEXaXfXfghinotuxy~uoigX@XDXDXFXKXKLMXNXUXUXWX]X]X_FW \\!Ah$ GHC"&q'*5RS\!]_cde_fgxz\]f_g!h5u JJu cd #DbD )_b1c_bV[2Ec%j2Q_b1cdhfhdh1i& uSz$B$'e:RSTVTaTacAUBCECPCPQlCmnpnznz{@nABDBKBKP]ghBKBijljqjqwzADjqjE!hLLD6\]^_b_j_jkoswrxyNqOlOTW_j`XY\YbYbcfYgJ'"-J #45dd;c!*0067@IJT@UV@Ur|rvw|4!8E UVfijopujvfwTYZdejZkTlm@UVNWXiNj!kNj@J@DEJ$(sSXz\]~mpqvw|q}m~[`arsxay[z"{Nj!kG tf* ++77 7q9G8M2dV!ks)S,18S1:8S7c tjdddd\}}|d}|dk(rtjtdz d 5}tj |d d|d dtj }d d d |jd}|j||d|jd|jd|jdn&|dk(r|j||jdddd}|d}tjj|dd} |j!|d d df|d d df|d d df| ddd} |j| |dd|jd|jd |jd!|j#d"|j%d#d$n8|d}tjj|dd} |j!|d d df|d d df| ddd} |d%k(rx|d&} tj&dt)|dz t+d't)|t,(D]4} | | d dfD]&} |j/|| | gdf|| | gdfd)d*d+,(6|j| |d|jdd-d.t)|d/d0d1||jd |jd"|j1d2t)|d3d4z}t3|d5D]V\}}|d6|t-|d7zd8d8|zf}|dj/tj4d|dz|d9d:|d;<X|djd=|djd>|djd?|dj7|dj9d@A|dj;dBdCdD|dj<dEdFdGdHdIJ|j?|dK|jAtB|dLz dMNtjD|y #1swYxYw)OzDShow the real prepared representation and the two condition vectors.r/ry) g@TrrrrpFrqrsdepth_mmNviridis)cmapzCAD depth (mm))rrz'real 2-D input: CAD view 0 depth pixelszpixel xzpixel yr3d) projectionrrrT)rsr rasterizedz(norm of prepared representation features ףp= ?)rrshrinkz)real 3-D input: sparse CAD surface voxels CAD X (mm)z CAD Y (mm) CAD Z (mm)irri&r z#607d8b ףp= ?g333333?)rlwrz*real point input: 6,000 CAD surface pointszreal graph input: rz CAD nodes + edges)rrrequalr*rwrrrrVog@z XLS process)markermsrz0condition actually supplied to every CAD elementz4process feature index (source names retained in NPZ)zfixed physical normalized value?rg{Gz?r-z^local provisional heater values: zero-imputed local heater validity mask: false for these runsrz#fff2ccz#d99b00g?) facecolor edgecolorr) transformrbboxz1: prepared input and XLS/heater condition summaryz.pngr)#rrr3rr r=rQimshowcolorbarrrrdelaxes add_subplotrrscatter set_zlabel view_initrr:rcrplot set_aspectrrhrrtext transAxesrrrr)rrrrrr~imageshownrfeaturedrawrnode neighbour process_widthr<rrHs rprocessed_input_figurers Q9NIC aB z WWTIIX] ^bhHHVL1!4fZ6H6KRVVTE_ %i 0#,,uSc,2d >?yAY[][h[hir[s   BCOOAaTOB N299>>$7P2QWX>#Y7 ZZAaCQqS#ac(Wqi[_Z ` TR&PX[ \ @A"--P\B]^`^k^klx^yz|{H{HIU{VWYWcWcdfgjWk8n td>?|j9d@dA|dd>?|j3tj4dKdddLdM;|j7dNdIdOd>d>?|j9d@dPdQdBdCiD|j7dHdRdSd>dTU|jtdVz d0dWXtj|ycc}wcc}wcc}wcc}w)YNTrry) rrtr^r`z denoise MSE)rrazrisk BCErrg?rr)rzCFour conditional diffusion denoising losses (weak-label diagnostic)zloss_comparison.pngrr})r0rrmrrro)rrvrqrvzgenerated deformation |mm|rrz&weak risk / unvalidated run-level typemagmar/r)rrrvminvmaxrr rrgffffff?)rrz; fixed CAD projection and colour scales; generated sample 0z_four_variant_comparison.pngr)rg @)roff)z2D CAD views 4 pixel messagesz!3D sparse voxels 6 voxel messagesz"6k CAD points kNN + global poolingz~6k CAD graph edge messagesg{Gz?rg= ףp=?rz#e8f0fez#2855a6)fcecg ףp= ?gq= ףp?center)hava)g\(\?gq= ףp? arrowstylez->) arrowprops)g?rz#eaf6ecz#278343rcr]z,process_x + masks provisional heaters + mask)rrz#fff1e6z#ae5f15g= ףp=?u0DDPM signed d → CAD + normal × d risk overlay)rr])g333333?r]g{Gz?zeAll local heater projections are mask=false (heater_projection_valid=false); values are zero-imputed.z#8a4b00)r rzarchitecture.pngtight)r bbox_inches)rrrrziprr set_yscalerrrrrrrr3rrZrr rrrrUr add_patch Rectangler annotate)reportsrrrrrepr!r$colrrrxdispcloudrarrHrrhirCrxxrs rfiguresr3 sIIdDI)CLL1Vgk4l83tdjjl7+3 i.a!8a!G*a!8PQ9RPQ1!I,PQ9RYfghjhohoFGqHFGABqrszq{FGqH]^I_]^XYJKLRJS]^I_fphoiqrtrr@ErFGIGSGSTWXaTbGcdfdkdkrtdkduvxvvIJvvK,LLVWX[XcXcdgh}d}CFXcYGHKHQHQRUHV \\!Af EHCW%C ''#Y(77 8Q|_TUVbTcVijlHjIJNjOdhVYZ`aefghlflamZmVmPUtuvHtIJLtMos*3c$i@\]fgi5jlprZ[bcdle5f+g &C&d2CG}Brzz%!*U1Q3Z&1RV\]cequzvQwyxExEFMxNOQO[O[_bcl_m^nnpqvpw\xOyz|zGzGHTzUVXVcVcdpVqrur~r~@DFNPr~rQ+g&LLC5STUWZWbWbcfjminoKhLdLQTWbXUVYV_V_`cVd << )FC"''%.APSUS^S^_bcfghSiQR&M52<< r#hs2QZ^g0h#ijljqjqrtuxrxy|~CGOS[jqk\]_]h]hikluwyz}w}~AvBO[\`Na]h]bMLLyB)yIJ277SUVY[JNVZb7Lcdfdpdpqtq~q~HIKLNR[_hqidjkmkrkrsvwz{os{GkrkHIKITITUWXabkxDEIwJITIKLNLSLSTVWZ[BFNU^LSL_KK&&37KCCIIcN"99RqHI_s( Q : Q%  Q* 0 Q/ c:|dd}djd|dD}djd|dD}djd|dD}td z jd t|dd |d |d |d|d|d|d|d|ddy)Nrrvr"c3K|]]}d|dd|dd|ddd|dddd d|ddd d d|ddd d d|ddd d d|ddd dd_yw)zrz rruz.4frwrvrdz.3frerfrgrarbz Nr<rrs rrzgallery..!sDqClm8AiL>1_3E2FiPQR]P^_bOcclmnoxmyz|m}GnHILmMMVWXYbWcdfWghqWrsvVww@ABCLAMNPAQRpAqru@vv@ABK@LMO@PQm@norss|}~H}IJL}MNT}UVZ}[|\\fgqCsA#A%variantsu · c3:K|]}d|dd|ddyw)zz prepared inputNr<r6s rrzgallery.."s8HuGop5a l^59VijuGsc3FK|]}d|dd|dd|ddyw)z z2 attribution (JSON)Nr<r6s rrzgallery..#skp]oXY?) ~McdefodpcqrdefgpeqdrrRS]os!z index.htmlaKCAD conditional diffusion diagnostics

CAD + process conditional-diffusion representations

Weak-label diagnostic only. z NG run(s), registration and defect extents are unvalidated, and local heater projection is invalid. Type is an observed diagnostic label, never a validated classifier.

Prepared representation and condition inputs

z[

Denoising loss

Same z( CAD view and colour scale

z3
representationnodes/runbest lossz MAE mmzY RMSE mmCAD-correspondent Chamfer mmregistered-scan Chamfer mmz weak AP

Feature attribution diagnostics

z.

full report

rr)joinrrr:)reportrrowsinputs attributionss rgalleryr?svrC DqwxBqCD DD ;;Htz|FuGH HFp]cdn]oppL!!'~  B C I J P C Q  R ~ S S w x ~ w   Z[^Z__GHKGLLstwsxxHILHMMfgjfkk}~B}CCvwCvDDr#uF!Gr&c tttttfDcgc]}|j ddc}g}t D]}t|}t||t|||j|j|j|j|j}t||t|dz |j!|dDcgc] }t#|}}t%||ddd||d|d j'd d d d ddt)|dgdddddd}tdz j+t-j.|dddt1|tdz j+dd|Scc}wcc}w)NTrrlrrv:four_conditional_diffusion_weak_label_diagnostic_completedFzcCAD/process conditioning -> signed-normal deformation diffusion -> p=CAD+n*d plus weak risk overlayrunvalidated_run_level_labels)rLstatusz*registration_independently_validated=falsez)defect extent/identity not fully reviewedzKheater_projection_valid=false; local values zero-imputed with explicit maskzonly z* NG run(s) and one CAD; no held-out resultzgen_models/processed_inputszgen_models/modelszgen_models/outputszgen_models/feature_attributionsr)rCrkrr7common_contractrrcritical_limitationsoutput_directoriesz report.jsonryrrrz README.mdzGradient plots are local post-fit sensitivity diagnostics of the weak-label risk head. They are not causal process attribution or feature importance. )r rrrrrVARIANTSrrrr learning_raterrrOrrr7r3r8r:rrrr?) argsr r,rrr-r$rr;s rrr'sc#cRVEWXEWd;EWXbdZa T "$:4$E=Y]^bcgcncnospBpBCGCWCWX\XdXdeievevLwsyDEIJNORVZU[[^S_O_y`ahaoaopsatx. ).Q#a&.D ) GDH0lAFNR^ex]uyzLuMuTuTuV`~lXDEpq~A F G J K O G P F Q Q { | W} g D N a l @ X y S z '{ v""4::f%q#Q[b"cdklrdsuy{FvFuRuRSlv}uRu~FLL Y *s FF c >tjt}|jdd|jdtd|jdt d |jd td |jd td |jdtd|j }|js tdt|}ttj|d|dDcgc] }|d|ddc}dddycc}w)N) descriptionz--allow-diagnostic-weak-labels store_true)actionz--epochs)typedefaultz--learning-rategMb`?z--diffusion-stepsrz--sample-stepsrz --samplesr}zjPass --allow-diagnostic-weak-labels: paired registration, defect extent and heater projection are invalid.rCr7rru)rr)rCr7Fryr) argparseArgumentParserr* add_argumentrr parse_argsallow_diagnostic_weak_labels SystemExitrprintrr)r rIr;r$s rmainrX0s'2A1>>Bbjv>3wxyyGyGHRX[dfyGyghihvhvwHNS\`hvhabcbpbpqDJMVXbpbYZ[ZhZhiyBKMZhZNOPO]O]^ior{|O]O}CDCOCOCQ~B  , ,J@l5m/m t9FU4::x0@CIJTCUMVCU}~UVW`UaijkviwMxCUMV'WejrstuMVs7D__main__)r$r$r!z torch.dtyper%r)r@zdict[str, np.ndarray]r%ztuple[np.ndarray, list[str]])r$r$r\r$r%r$)rjr$rkrr%r$)rr7r%zEtuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray, dict[str, Any]])r%z'list[tuple[str, dict[str, np.ndarray]]])rr7r%dict[str, Any])r/rr%r)r7r$r8r$r%z float | None)r<rr%r7)rr$r@r$r%r) rMrrrrNrrOrrPrr%r)rr7rrZrrrXrrrrrrOrr%rZ)rr7rrZrxrr%r&)rr7rrZr%r&)r,zlist[dict[str, Any]]rr7r%r&)r;rZr%r&)rIzargparse.Namespacer%rZ)r%r&)Ar* __future__rrQhtmlrrpathlibrtypingr matplotlibusematplotlib.pyplotpyplotrrr3 scipy.spatialrr"rr torch.nn.functional functionalr models.datar r r models.trainingr rcParamsrrrrrrrrGr4r%rNr^rnrrrModulerr0r:r=rDrrUrrrr3r?rrXr'r<r&rrjs=#!! u! LL+  =PUVW \p.opS#t  ?05}}+ Z>#$NJ4EbiiE,]I *y w M4#>LS$GLu Zr&