+ j<^RIt^RIt^RIHt^RIHt^RIH t ^RI H t H t ^RI HtHtHtHtHtHtHt] !4'd^RIHt] !4'd^RIHt!RR]4t!R R ]P44tR R ltR#) N)!_is_torch_distributed_initialized)is_scipy_availableis_vision_available)HungarianMatcher _set_aux_lossbox_iou dice_lossgeneralized_box_iounested_tensor_from_tensor_listsigmoid_focal_loss)center_to_corners_format)linear_sum_assignmentcNa]tRt^(to]P !4R4tRtVtR#)LwDetrHungarianMatcherc  VR,PR,wrEVR,P^^4P4pVR,P^^4p\P!VUu.uF qR,NK up4p \P!VUu.uF qR,NK up4p Rp Rp ^V , Wl,,^V, R,P 4),p V ^V, V ,,VR,P 4),pVR V 3,V R V 3,, pVP pVP\P4pV P\P4p \P!Wz^R 7pVPV4p\\V4\V 44)pVPV,VPV,,VPV,,pVPWER 4P!4pVUu.uFp\#VR,4NK pp.pWS,pVP%V^R 7p\'V4FpVV,p\)VP%VR 44UUu.uFwpp\+VV,4NK pppV^8XdTpKV\-VV4UUu.uF`wpp\.P0!V^,V^,VV,,.4\.P0!V^,V^,.43NKb pppK VUUu.uFRwpp\P2!V\P4R 7\P2!V\P4R 73NKT upp#uupiuupiuupiuuppiuuppiuuppi)zz Differences: - out_prob = outputs["logits"].flatten(0, 1).sigmoid() instead of softmax - class_cost uses alpha and gamma logits:NN pred_boxes class_labelsboxesg?g@g:0yE>NNN)pdim)dtype)shapeflattensigmoidtorchcatlogrtofloat32cdistr r bbox_cost class_cost giou_costviewcpulensplitrange enumeraterzipnp concatenate as_tensorint64) selfoutputstargets group_detr batch_size num_queriesout_probout_bboxv target_ids target_bboxalphagammaneg_cost_classpos_cost_classr'rr&r( cost_matrixsizesindicesgroup_num_queriescost_matrix_listgroup_idgroup_cost_matrixic group_indicesindice1indice2js &&&& _/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/loss/loss_lw_detr.pyforwardLwDetrHungarianMatcher.forward)sI#*("3"9"9""= 8$,,Q2::<<(00A6YY7C7a. 1 17CD iiW =W7W => e)8a(lT>Q=V=V=XVW#AzM2^AzM5RR ;;u}}-!nnU]]3 KK; LL' ))A()KMefqMrss nny04??Z3OORVR`R`clRll !&&zCGGI *12'QQwZ'2'5&,,->A,Fj)H 0 : ENO`OfOfglnpOqErsErTQ21Q48ErMs1}'-0,G  -H( GAJARU]A]4]'^_ GAJ'?@-H  *lsskrcgcdfg%++6QVQ\Q\8]^krssS D =.3 tts&1N3N8;N=*O%A&OAON) __name__ __module__ __qualname____firstlineno__r no_gradrQ__static_attributes____classdictcell__) __classdict__s@rPrr(s ]]_6t6trcaa]tRt^ctoV3RltRt]P!4R4tRt Rt Rt Rt Rt R tR tVtV;t#) LwDetrImageLossc`<\SV`4WnW nW0nW@nWPnR#N)super__init__matcher num_classes focal_alphalossesr7)r4rcrdrerfr7 __class__s&&&&&&rPrbLwDetrImageLoss.__init__ds)  && $r\c RV9d \R4hVR,pVPpVPV4p\P!\ W#4UU U u.uFwpwrVR,V ,NK up p p4p VP p ^p VR,V,p\P!\ W#4UU Uu.uFwpwrVR,V,NK upp p^R7p\P!\\VP44\V44^,4pVPV4pVP4P4pVP4p\P!V4pVPV 4PV4pW{3,pVV,PV 4VP^V , 4,p\P !VR4P4PV4pVVV&^V, VV&V)VP#4,V^V, P#4,, pVP%4V, pRV/pV#uup p piuupp pi)rz#No logits were found in the outputsrrrrg{Gz?loss_ce)KeyErrorr_get_source_permutation_idxr r!r/rediagrr detachr#cloner zeros_likepowclampr"sum)r4r5r6rE num_boxes source_logitsridxt_Jtarget_classes_or?r@ src_boxesrJ target_boxes iou_targetspos_iousprob pos_weights neg_weightspos_ind pos_qualityrjrfs&&&&& rP loss_labelsLwDetrImageLoss.loss_labelsms 7 "@A A) ##..w7 99CPWLa%bLayq&1a&7&:&:La%bc  L)#. yyWAV!WAVIAv!G*Q--AV!W]^_ jj ,Y-=-=-?@BZ[gBh ijk l "nnU+ $$&--/$$&&&}5 hhuo((/ ++7m''.a%i1HH kk+t4;;=@@G * G ; G,+kQXNN 55g> |, #/ %,-W7W-*LwDetrImageLoss.forward.. s@1An-..s!)rr)op)minrrx)r7_enc)trainingr7itemsrcrsr r2rnextiterrrrdist all_reduceReduceOpSUMget_world_sizerritemrfupdaterr.)r4r5r6r7kr<outputs_without_aux_and_encrErt world_sizerfrrJrl_dictrs&&& rPrQLwDetrImageLoss.forwardsR)- T__1 $]]_' ,TQ]0BDqL_G_DAD_ $' ,,:ZP@@@ * OOYKu{{4PTU\UcUcUePfKgKnKno  , . . OOI$--*;*; <,,.JKK 6A>CCE KKD MM$--wS T  ' )(1':M2N(O$,,'8:N KKDw !]]4GV_`F9?HaAaS'k1nFHMM&) ()P G #!-0Kll;JlOG t'IV4:LLNCNDA!f*a-NC f%$  U' >IDs J7J7 J7<J= K)rer7rfrcrd)rTrUrVrWrbrr rXrrrrlrrrQrYrZ __classcell__)rgr[s@@rPr^r^csO%"H ]]_   2D%% D66r\r^c aa\VPVPVPR7p .R Op \ V VP VP V VPR7p V PV4/p RpW R&W=R&RVRV/V R&VP'd\WV4pWR&V !W4oRVPRVP/oVPSR &VP'dv/p\VP^, 4FBpTP!SP#4UUu/uFwppVR V 2,VbK upp4KD SP!V4SP#4UUu/uFwppVR ,VbK pppSP!V4\%VV3R lS44pVSV3#uuppiuuppi))r'r&r()rcrdrerfr7Nrrrrrjrrrxrc3b<"TF$qS9gK SV,SV,,xK& R#5ir`rS)rr loss_dict weight_dicts& rPr/LwDetrForObjectDetectionLoss.._s)Ti CS,y|k!n,,is//)rrr)rr'r&r(r^ num_labelsrer7r#auxiliary_lossrclass_loss_coefficientbbox_loss_coefficientgiou_loss_coefficientr-decoder_layersrrrs)rrrrconfig outputs_class outputs_coordenc_outputs_classenc_outputs_coordkwargsrcrf criterion outputs_lossraux_weight_dictrJrr<enc_weight_dictrrrs&&&&&&&&&, @@rPLwDetrForObjectDetectionLossr.s%$$0@0@FL\L\G0F%%&&$$ ILLL#!+#'#L)-G,=(),/If;;[&JfJfgK%;;K  v,,q01A  " "{?P?P?R#S?Rtq!A!A3KN?R#S T2?+1<1B1B1DE1DAq6z1}1DOE' TiT TD - -- $TEs *F8 3F>)NNNN)numpyr0r torch.distributed distributedrtorch.nnrdistributed.utilsrutilsrrloss_for_object_detectionrrrr r r r transformers.image_transformsr scipy.optimizerrModuler^rrSr\rPrsl A;F48t-8tvHbiiHV2.r\