jdadZddlmZddlZddlZddlZddlZddlmZddl m Z ddl Z e jdddl mZddlZddlZddlmZmZddlmcmZdd lmZmZmZdd lmZed z Zd Zej@fdd Z!ddZ"d dZ#d!dZ$GddejJZ&d"dZ'd#dZ(d$dZ)d%d&dZ*ejVd'dZ,d(dZ-d)dZ.d*dZ/d+dZ0d,dZ1e2dk(re1yy)-aConditional DDPM diagnostic for CAD/process -> deformed scan point clouds. The model diffuses the CAD-node signed normal deformation, not free 3-D point positions. A generated point cloud is reconstructed as ``CAD + normal * d``; this keeps every generated point traceable to its source CAD node and makes a node-level risk/defect overlay possible. The two supplied AC28 runs are deliberately treated as a weak-label diagnostic: their PLY-to-CAD registrations, defect extent, and heater-zone projection have all been marked invalid in the prepared artifacts. The command cannot claim production performance or held-out generalization. ) annotationsN)Path)AnyAgg)Tensornn)ROOTfixed_physical_normalizesplit_embedded_units)seed_everythingzresults/models/diffusiong>@c0tj||S)Ndtype)torch as_tensor)valuers *gen_models/models/conditional_diffusion.py_tensorr&s ??5 ..ctj|dtjj}tj|dt}tj|dj t }t|\}}|j}tjt|t}d|d|dfd|d |d ffD]M\}}} td d D]8} || d } | |vs| | d z s|j| } || d z || <d|| <:O||z} ttj| |ddtj||d}tj||j tj|j tjgj tjS)zGNormalize global XLS values and append observed/imputed validity masks. process_xprocess_feature_validprocess_feature_namesru 상부온도 heater_top_cheater_top_imputedu 하부온도heater_bottom_cheater_bottom_imputedr 02dTrN)npasarrayfloat32copyboolastypestrr tolistzeroslenrangeindexr where concatenate)itemrawobservednames base_namesunitsimputedprefixvalues imputed_zoneszonekeyr,usable normalizeds r_process_vectorr=*s **T+& 3 8 8 :Czz$67>H JJt34 5 < >:xrzz'BGNNSUS]S]D^_ ` g ghjhrhr ssrc\ttdz jd}t|dk7rt dt|g}|D]}t j |d5}|jDcic]}||| }}dddd}|Dcgc] }|vs| }}|rt |jd ||j|ft|d d d jd tfd |Dr t d|dScc}w#1swYxYwcc}w)zJLoad both paired artifacts without promoting their weak targets to labels.zdata_processed/pairedz*_cad_ply_targets.npzzzload_data..Ss+ GwGAt4  # #A &% /ws"%z(3co3(> $))IgY?@ @d|$  1 l+11!4 5E Gw GGYZZ 77>/ . ?s*&D5 DD D)D)DD& crtj|dtjdz }tj|dtj}ttj|dj }tj|dtj}tj|dtj}tj|dtj}tj |||gd d z }tj t|dft|tj }||z}t|} tj| d t|d } tj||||| gd tt| gd|| dfS)zOBuild per-node condition features using values only when their masks permit it.rA@@rBrHrG(provisional_heater_bottom_c_per_cad_node'provisional_heater_delta_c_per_cad_noder axisg@@rNr)top_cbottom_cdelta_c)global_process_features!local_provisional_heater_channelsrH&local_values_zero_imputed_when_invalid) r!r"r#r%r/stackfullr*floatr=repeatr.rX) r/posnormalprojection_validtopbottomdelta local_values local_maskglobal_processglobal_per_nodes r _conditionsrzXs^ **T,' 4v =C ZZ\*BJJ 7FBJJt,E'FGLLNO **TABBJJ OC ZZGH"** UF JJtEF SE88S&%0q9EAL#c(A.>(?rzzRJ*,L$T*Niit 4c#hQGO >>3 j/RYZ [#&s>':#;-M#36F2F ^ rcBeZdZdZddfd ZedddZd dZxZS) ConditionalDisplacementDDPMzISmall node-wise conditional epsilon model with a deterministic risk head.c `t|tjtj||tj tj||tj |_tjtjd|tj tj|||_tjtj|dzdz|tj tj||tj tj|d|_tjtj||dztj tj|dzd|_ y)Nr?r ) super__init__r SequentialLinearSiLU conditiontimedenoiserrisk)self condition_dimhidden __class__s rrz$ConditionalDisplacementDDPM.__init__qs. ryy'GTVT]T]^dflTmoqovovoxyMM"))B"7BIIfV\D]^  bii Q&GTVT]T]^dflTmoqovovoxz|{D{DEKMN{OP MM"))FFaK"@"'')RYYW]abWbdeMfg rc B|dz}tjtjdtjd||j |j }|dddf|dzdz }tj|j|jgdS)Nr?rra)devicergI@r dim) rexplinspacemathlogrrcatsincos)trhalffreqphases rtime_embeddingz*ConditionalDisplacementDDPM.time_embeddingxs{axyy488F+;T!((Z[ZaZabc!T' T$Z'$.yy%))+uyy{3;;rc&|j|}|j|j|}|jt j |dddf||gdj d}||j|j dfS)Nr r)rrrrrrsqueezer)rnoisy_displacementtimesteprencodedrepsilons rforwardz#ConditionalDisplacementDDPM.forwards..+yy,,X67-- +=ag+FQU*V\] ^_gghij '*221555r))rrXrrXreturnNone)r~)rrrrXrr)rrrrrrrztuple[Tensor, Tensor]) __name__ __module__ __qualname____doc__r staticmethodrr __classcell__)rs@rr|r|os%Sh<< 6rr|cttjdd|}d|z }tj|d}|||dS)Ng-C6?{Gz?r rr)betaalphaabar)rrcumprod)stepsrrrs r _schedulers: >>$e ,D HE ==A &D5$ 77rcbt|j}|dk(rytj| d}||j tj }tj |tjdt|dzz }t||zj|z S)Nrstable)kindr ) rXsumr!argsortr&float64cumsumaranger*rn)scoretarget positivesordery precisions r_aprsFJJL!IA~ JJvH -Eu RZZ(A ! ryyCFQJ77I )a-$$&2 33rc |dk\}|jt}t||zj}t |||t dt||zjz d|zt dt|jt|jzz t|jt|jdS)N?r r?)APz IoU_at_0.5z Dice_at_0.5predicted_positive_nodestarget_positive_nodes)r&r%rXrrmax)rrpredr intersections r _risk_metricsrs B;D dAq~~'(LeQ-|c!S$QR(IYEZ>[/[|+c!S_s1557|5S.TT(+DHHJRUVWV[V[V]R^ ``rcdfd }t||t||}}tj|d|djd}t |j dj j|j dj jzjdz S)zKSymmetric point-set Chamfer in millimetres, with deterministic subsampling.ct|kr|S|tjdt|dz tjS)Nrr r)r*r!rint64)a max_pointss rsamplez_chamfer..samples: q6Z HQA JbhhGHHrNrr r?)r np.ndarrayrr) rrcdistrrnminr7meanr/)rrrrrbdistances ` r_chamferrsI 6$< '&."9qA{{1T7AdG,44Q7H (,,q/((--/(,,q/2H2H2M2M2OOUUWZ[[ \\rc "tjj|}tjt ||}t j t jdt |ddz |t jddd}t|D]\}} tjt |ft| tj} ||| |\} } |d| } |d| z j| zz | jz }|t |dz k(rdnt||dz}|dkr|n5|d|j|zd|d|z j| zz}|jddS)zIEta=0 DDIM path; different initial seeds produce diagnostic alternatives.) generatorrrr rN)r Generator manual_seedrandnr*r!uniquerr enumeratermrXlongsqrtclamp)modelrschedulerseedrxsequenceicurrentrrrLrx0previouss r sample_ddimrsh!--d3I C Ni8AyyQHV,<(=(A5PRPXPXYZ[_]_[_`H) 7 JJAy#g,ejj A1a+ (1t8//#g-- <c(ma//2S!a%5IQ,BHV$4X$>$C$C$E$JaRZ[aRbckRlNlMrMrMtw~M~$~ * 772q>rc 4tjd\}}|jddddddd d d |d fd dg}|D]W\}}}}}|jtj||f||ddd|j ||dz z||dz z|dddYdD]\} } |j d| | dddd |j ddd ddd!"|j dd#d$dd%d&"|jtd'z d(d)*tj|y)+N) g@)figsizeoff)rg= ףp=? ףp= ?HzG?zCAD nodes position + normal)r333333?rrzXLS process_x values + masks)?r皙?rz,Local provisional heater values + valid maskrg ףp= ?rQ?zcondition MLP z features)\(\?g(\?zG?Q?uDDPM εθ noisy signed d, t)rg{Gz?rrz!Risk head weak defect probabilityz#e8f0fez#2855a6?) facecolor edgecolor linewidthr?center )havafontsize)))rffffff?)r皙?))rr)rgGz?))?r)rr))(\?r)rr))rr)rrz->z#415a77g?) arrowstylecolorlw)xyxytext arrowpropsgQ?gQ?usample d → CAD + normal × dz#172033)rrr rz_Diagnostic only: local heater mask is false because heater_projection_valid=false in both runs. z#8a4b00architecture.pngtight)dpi bbox_inches) pltsubplotsre add_patch RectangletextannotatesavefigOUTclose) rfigaxboxesrrwhlabelstartends r_architecture_figurer&sHll9-GCGGEN==M c3}oYGH<B  E#1aE S]]Aq61a9PYehij AE 1q1u9eXPRS#J s B3utU^eh9i jJGGD#7HrYbGcGGBv|DOPXaGbKK((cwKGSVrc |d}|d}tjdddd\}}dD]>\} } |d j|D cgc]} | d  c} |D cgc]} | |  c} | @|d jd |d j d |d j d |d j |d jd|dj|D cgc]} | d  c} |D cgc]} | d c} d|dj d |dj d|djdd|djd|jd|jtdz dtj|tjt|dddt|zfd\}}t|dk(r|d}t|D]\} \} }|d|d}}|||| |z| dz|zdftzzz}|||d | |z| dz|zdftzzz}|t!j"|dd d df|t!j$|| |z| dz|ztzd!d"d tf|t!j$|d | |z| dz|zftzd#d"d tf||| |z| dz|zd$d%d dfg}t'|| |D]\}\}}}}}}|j)|ddd f|dddf|d&|||d'}d(|vr'|d)k\}|j)||d f||dfd*d+d,d-d./|j+|d0j-d1||j/d2|j d3|j d4|j1||d56|jd7|jtd8z d9tj|ycc} wcc} wcc} wcc} w):NrOrPr r?) g @T)rconstrained_layout))totalr*)denoisez denoise MSE)rzrisk BCErepoch)r#rzdiagnostic lossr)rrisk_APz#8b1e3f)r ztraining weak-label APg{GzgRQ?z/Conditional-diffusion diagnostic training trace training.png)rgffffff@rArBz CAD baselineGreyszweak residual targetviridiszgenerated sample 0zgenerated + risk highlightmagmar)cscmapvminvmax rasterizedrrnonez#00d4ffg333333?u risk ≥ 0.5)r6 facecolors edgecolors linewidthsr#run_idz: equalz CAD X (mm)z CAD Z (mm)gRQ?)rshrinkzMCAD-correspondent deformed point clouds; weak/unvalidated training diagnosticpointcloud_comparison.pngr)rrplot set_yscale set_xlabel set_ylabellegendgridset_ylimsuptitlerrrr*rDEFORMATION_SCALE_MMr!r)abszipscatter set_titler/ set_aspectcolorbar)datasamplesr target_signedhistoryrOnraxesr:r#rowr,rLr/rprq target_pts generatedfieldsrcloudrtitler7lohidrawhots r_plotsrcs9oG _A Q9NIC\ U Q g6gsc'lg6W8UWcSW8U]b c]GutAw11':DGAUUWkmvxy|P Q U1WeAgq[13OQXZ[]^ _   8;4;7O 3B3udB::eAqDk5A;%3TXZacpt:uDrk 5a=%Q-1\ert}K L LLDN//12"UG< =r}}W?UWYWdWdeqWrtvuBuBCOuP LL"SL 1 8P/"LL`aKK11sK;SYYs^778U24Ws O4  O9 O>2 P ctt}|d}|d}ggggf\}} } } g} |D]\} }t|\}}|j|| jt j |dt t t z | j|ddddf| j|ddddf| j|tt j|}tt j| }tt j| }t j| jt}t|jd}t|}tjj!|j#|d}|j%j'd}t)||z |z j+d }g}t-d }d}d }t/|D]}|j1|j3d tj4d |t)|ftj6}tj8|}|d|} | j;|zd| z j;|zz}!||!||\}"}#t=j>|"|}$t=j@|#||}%|$d|%zz}&|&jC|jEtjF|#jIjK}'t-|dzt-|&jIt-|$jIt-|%jIt-tM|'|xsdd}(|j|(|(d|ks|(d}|dz}|jOjQD)*cic]$\})}*|)|*jIjS&}})}*| tUd|jW||jYtjZt/|D+cgc]}+t]||||d|+zc}+jK},tj^5|tj`|tjbt)|tj6|\} }#dddtjF#jK}-|jK}.g}/g}0te|D]\}+\} }tg|+|z|+dz|z}1|d|d}3}2|,dd|1ft z}4|.|1t z}5|4jid }6|4jkd }7|2|3|6dddfzz}8|2|3|5dddfzz}9|d}:|/jtm|djot-t jp|6|5z jit-t j:t jh|6|5z dzts|8|9ts|8|:t-|7jit-|7jutw|-|1||1t|djod |0j|2d|3d|4dddddfzzty|jdt{||,|-|.|t|jd d t jt|d!z t j|D cgc]!\} }tm|djo#c}} |d dd|d dd|,j|t)||t zt jZ|0d"|-jt)||t jd#d#gt jd$t j|D cgc]!\} }t|djo#c}} % t|d&z };tj|jO|jd|t d'd(d)|;d*d'd'd+|D cgc]!\} }tm|djo#c}} dd,| d-d.gd/d0d1d2d3d#d$d4d5||||||d d|||d6 |/d7d8d9d:gd;gd<d=}z jtj|)lr weight_decay)rinfrT) set_to_nonerr) pos_weightgffffff?g)r,r*r+rr-r*z%No diffusion training epoch completedi3(5)rrArBscan_pos_cad_mmr@r?rH) runsigned_normal_MAE_mmsigned_normal_RMSE_mm,node_correspondent_reconstruction_Chamfer_mmraw_registered_scan_Chamfer_mm)mean_stochastic_signed_deformation_std_mm(max_stochastic_signed_deformation_std_mmrrH)parentsexist_okgenerated_pointclouds.npzrdu찍힘,constant_diagnostic_unlearnable_single_class) r@rArBgenerated_signed_normal_mmgenerated_pointcloud_mmrisk_probability defect_typedefect_type_statusrH#conditional_diffusion_diagnostic.ptFz>two weak/unvalidated NG runs; not a validated predictive model) state_dictrdiffusion_stepsdeformation_scale_mmproduction_validwarning5conditional_diffusion_weak_label_diagnostic_completedzconditional DDPM epsilon predictor over CAD-node signed normal deformation; point = CAD position + CAD normal * generated displacementz8normalized XLS process_x with observed and imputed maskszposition_mm / 1000rq)global_process_x local_heater CAD_geometryu1cad_to_scan_normal_residual_mm clipped to ±30 mmz"10 mm weak defect probability/maskuboth runs are 찍힘)signed_normal_deformationrtypez?two supplied runs have one observed defect class and no OK data)rstatusreason) epochs learning_rater~ sample_stepsstochastic_samples loss_initial loss_bestselected_epochrVzHCAD-node signed-normal displacement against the weak registered residualz2generated vs CAD-node weak residual reconstructionzigenerated CAD-node cloud vs subsampled registered fused scan; includes sampling and registration mismatch)MAE_RMSErorp)zDregistration_independently_validated=false for both paired artifactszTdefect_extent_observed=false; risk target is a weak geodesic dilation around a pointzpheater_projection_valid=false; local provisional heater values are zero-imputed and their false mask is suppliedz=two NG runs and one CAD prevent a run/CAD held-out evaluationu4defect type is constant 찍힘 and cannot be learned)rr.rCrur|)rrgeneralization_measuredrrunsindependent_samples conditioningtargetsrztrainingmetricsmetric_interpretation limitationsoutputsz report.json ensure_asciiindentutf-8encoding)Hr r_rzrWr!cliprLrr.r&r%r|rJrroptimAdamW parametersr clamp_minr*rrnr+train zero_gradrandintr randn_likerFmse_loss binary_cross_entropy_with_logitsbackwardstepsigmoiddetachnumpyrr}itemsclonerSload_state_dictevalrlrno_grad zeros_liker)rslicerstdr'r/rMrrrr&rcrmkdirsavez_compressedr"reshapesave write_textjsondumps _write_html)=rrr~rTrrSrOrWcondition_parts target_parts risk_parts binary_partsmetadatarLr/rcondition_metar risk_target binary_targetrr optimizerpositiverjrV best_loss best_state best_epochr,rnoisernoisypredicted_noise risk_logitr+ risk_lossloss probabilityrYr:rr,r[ry target_npper_runcad_point_samplesslrprq generated_mm target_mmr uncertaintypredicted_cloud target_cloud scan_cloud checkpointreports= rrlrls  ;D$y/tO7L1>@"b"n;O\:|H4$/$5! >y)BGGD)I$JMaLacwx|PP Q$89!Q$?@D!34QT:;' 78I72>>ZfKgCh&xACAOAOPZA[y\junpn|n|}InJnQnQRVnW^k ' (: ;E)H !!%"2"2"4UY!ZI **1-H{#h.(:AAcAJJ&(GeEl)koJ~q{v y**t*<==Oc&k^5::V  (&1A(1K$ f$D'85'@@&+E8Y&G#**_e466z;[ef y( )mmJ/668>>@ eai(53GTYZaZhZhZjTkuz|E|L|L|NvO\abefqs@bAbHEH\IJs w<) #G I519jHMHXHXHZH`H`HbcHb*#u#u||~3355HbJc BCC *%uzz| w|~ExFGxFns[ 8\X`chXhixFGHNNPI e..v6 CKW\WaWa8bdmn : }}Z0668 I$&G%g.y4 519uqyAo .d<>PRVWcRdV B'*>> bM$88   #<3C3CA3F[ag 66Vi4&888 +, tH~**,-$)"&& 1A*B*G*G*I$J%*277277D9{?O?O?Q9R8=koo>O8P!"22"6 b8IJ'+D1J,K,P,P,R'S     TVD\LAt7ST:VW=YeKf3<3D3DWcRYl\]3^au3u029JQR0S)9)A)A#g,PQ)R$&JJ(/C$DY[YcYceSZT02 x)IK1doS]st!)5cpqr% Zi|H+2GAJwDWfoDN[bc.xk_Cn"oeYF =$$TZZUST%U`g$h M}dG :+^ rlz rmz.3frnrorprrNzn/az )htmlescape)rKrYs rrNz_write_html..Ws*sarZ]Xdkk#e*56iDZ@[\_?``ijmoFkGHKjLLUVYZHVIJMUNNWX[\|X}~AWBBKadekalmqara~LOPVLWX\L]DIKJJTUarsAArz index.htmluConditional CAD diffusion diagnostic

CAD + process → conditional deformed point-cloud diffusion

Weak-label diagnostic only. Two NG runs, unvalidated registration, provisional heater projection, and a single observed defect class (찍힘) mean this is neither a validated predictor nor a generalization result.

The DDPM generates signed normal CAD-node displacement, then reconstructs each point as CAD position + CAD normal × displacement. Risk is shown at CAD nodes; defect type remains the constant diagnostic value 찍힘 because it is unlearnable from one class.

Architecture

Conditional diffusion architecture

Training

Denoising loss curve

Generated point clouds and risk highlight

CAD target and generated point clouds

Weak-target fit metrics (mm)

u
RunMAERMSECorrespondent ChamferRaw scan ChamferWeak risk AP

full report · generated point clouds · diagnostic checkpoint

rr)joinrr)rrowss rrrVss 77saghqars sD<##)} ~B}CCY %\ gn $orc tjt}|jddd|jdtd|jd t d |jd td |jd td|jdtd|j }|js tdt|j|j|j|j|j}tt!j"|d|dd|d|ddddy)N) descriptionz--allow-diagnostic-weak-labels store_truez:required because current labels are explicitly unvalidated)actionhelpz--epochsP)rdefaultz--learning-rategMb`?z--diffusion-steps2z --samplesz--sample-stepsz`Refusing unvalidated targets. Pass --allow-diagnostic-weak-labels to run the two-run diagnostic.rrrrrz)rrrrzFr?r)argparseArgumentParserr add_argumentrXrn parse_argsallow_diagnostic_weak_labels SystemExitrlrrr~rTrprintrr)parserargsrs rmainr`sG  $ $ 9F 8UQR  b9 )tD +#rB  #q9 (sB?    D  , ,{|| d00$2F2F VZVgVg hF $**x 0vj?QR]?^kqr{k|NTUbNcdsxAB CDr__main__)rrrz torch.dtyperr)r/dict[str, np.ndarray]rr)rdict[str, Any])r/rrz!tuple[np.ndarray, dict[str, Any]])rrXrdict[str, Tensor])rrrrrz float | None)rrrrrzdict[str, float | None])i)rrrrrrXrrn) rr|rrrrrrXrrXrr)rrXrr) rSrrTrrrrUrrVzlist[dict[str, float]]rr) rrXrrnr~rXrTrXrrXrr)rrrr)rr)3r __future__rrrrrpathlibrtypingr matplotlibusematplotlib.pyplotpyplotrrr!rrrtorch.nn.functional functionalrrSr r r rr rrLr#rr=r_rzModuler|rrrrrrr&rcrlrrrrrr s #  u FF% ''5:MM/t.8..6"))6.84`]  X* LF`Fo D zFr