^jY<ddlZddlZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZmZe rddlmZe rddlmZGdd e ZGd d ej0Z d d Zy)N)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_assignmentc:eZdZejdZy)LwDetrHungarianMatcherc v|djdd\}}|djddj}|djdd}tj|Dcgc]}|d c}} tj|Dcgc]}|d c}} d } d } d| z || zzd|z d zj z} | d|z | zz|d zj z}|dd| f| dd| fz }|j }|jtj}| jtj} tj|| d }|j|}tt|t|  }|j|z|j|zz|j|zz}|j||d j!}|Dcgc]}t#|d}}g}||z}|j%|d}t'|D]}||}t)|j%|d Dcgc]\}}t+||}}}|dk(r|}It-||Dcgc]E\}}t/j0|d|d||zzgt/j0|d|dgfG}}}|Dcgc]O\}}tj2|tj4tj2|tj4fQc}}Scc}wcc}wcc}wcc}}wcc}}wcc}}w)z Differences: - out_prob = outputs["logits"].flatten(0, 1).sigmoid() instead of softmax - class_cost uses alpha and gamma logitsNrrr pred_boxes class_labelsboxesg?g@g:0yE>)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 i/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/loss/loss_lw_detr.pyforwardzLwDetrHungarianMatcher.forward(s=#*("3"9"9"1"= K8$,,Q2::<<(00A6YY7Ca. 1CD iiW =7 => e)%8a(lT>Q=V=V=XVW#AzM2^AzM5RR ;;u}}-!nnU]]3 KK+; LL' ))A()KMefqMrss nny04??Z3OORVR`R`clRll !&&z;CGGI *12QQwZ22':5&,,->A,Fj) H 0 : ENO`OfOfglnpOqErsTQ21Q48sMs1}'-0,G  ) GAJARU]A]4]'^_ GAJ'?@ lsscgcdfg%++6QVQ\Q\8]^ssS D =.3 tts&! L LL$<L).A L/AL5N)__name__ __module__ __qualname__rno_gradrPrOrr'sU]]_6t6trVrcveZdZfdZdZej dZdZdZ dZ dZ dZ d Z xZS) LwDetrImageLosscht|||_||_||_||_||_yN)super__init__matcher num_classes focal_alphalossesr6)r3r]r^r_r`r6 __class__s rOr\zLwDetrImageLoss.__init__cs3  && $rVc vd|vr td|d}|j}|j|}tjt ||D cgc]\}\} } |d| c} } }} |j } d} |d|}tjt ||D cgc]\}\} }|d|c}} }d}tjtt|jt|d}|j|}|jj}|j}tj|}|j| j|}|| fz}||j| |jd | z z}tj |d jj|}|||<d |z ||<| |j#z|d |z j#zz }|j%|z }d |i}|Scc} } }wcc}} }w) Nrz#No logits were found in the outputsrrrrrrrg{Gz?loss_ce)KeyErrorr_get_source_permutation_idxrr r.r_diagr rdetachr"cloner zeros_likepowclampr!sum)r3r4r5rD num_boxes source_logitsridxt_Jtarget_classes_or>r? src_boxesrI target_boxes iou_targetspos_iousprob pos_weights neg_weightspos_ind pos_qualityrcr`s rO loss_labelszLwDetrImageLoss.loss_labelsls 7 "@A A) ##..w7 99CPWY`La%b%byq&1aa&7&:%bc  L)#. yyWgAV!W!WIAv1!G*Q-!W]^_ jj ,Y-=-=-?@BZ[gBh ijk l "nnU+ $$&--/$$&&&}5 hhuo((/ )++7m''.a%i1HH kk+t4;;=@@G * G ; G,+kQXNN 55g> |, #J/ %,-7-- c tjt|Dcgc]\}\}}tj|| c}}}}tj|Dcgc]\}}| c}}}||fScc}}}wcc}}wrZr)r3rDrIrqtargetrrs rOrz+LwDetrImageLoss._get_target_permutation_idxsiIIPYZaPbccna!Vuvq9cd YY'B;AvBC *$$dBrc|j|j|j|jd}||vrt d|d||||||S)N)labels cardinalityrrzLoss z not supported)r}rrr ValueError)r3lossr4r5rDrmloss_maps rOget_losszLwDetrImageLoss.get_losss^&&00____   x uTF.9: :x~gwCCrVc d|jr |jnd}|jDcic]\}}|dk7s |dk7s||}}}|j|||}t d|D}||z}t j |gt jtt|jj}d} tjrWtjrCtj|tj j"tj$} t j&|| z dj)}i} |j*D]&} | j-|j/| ||||(d|vrt1|dD]~\} } |j| ||}|j*D]W} | dk(r |j/| | |||}|jDcic]\}}|d | z|}}}| j-|Yd|vrv|d}|j||| }|j*D]N} |j/| ||||}|jDcic] \}}|d z| }}}| j-|P| Scc}}wcc}}wcc}}w) a This performs the loss computation. Args: outputs (`dict`, *optional*): Dictionary of tensors, see the output specification of the model for the format. targets (`list[dict]`, *optional*): List of dicts, such that `len(targets) == batch_size`. The expected keys in each dict depends on the losses applied, see each loss' doc. r enc_outputsauxiliary_outputsc38K|]}t|dyw)rN)r*).0rps rO z*LwDetrImageLoss.forward..s@1An-.@s)rr)op)minrrq)r6_enc)trainingr6itemsr]rlrr1rnextiterrrdist is_availableis_initialized all_reduceReduceOpSUMget_world_sizerkitemr`updaterr-)r3r4r5r6kr;outputs_without_aux_and_encrDrm world_sizer`rrIrl_dictrs rOrPzLwDetrImageLoss.forwards)- T__1 $]]_' Q]0BqL_G_AqD' #' ,,:GZP@@@  * OOYKu{{4PTU\UcUcUePfKgKnKno     4#6#6#8 OOI$--*;*; <,,.JKK J 6A>CCE KK UD MM$--gwS T U ' )(1':M2N(O *$$,,'8':N KK*Dw !]]41BGWV_`F9?HAaAaS'k1nHFHMM&) * * G #!-0Kll;JlOG  &t['7IV4:LLNCDAq!f*a-CC f% &  U' >IDs J J J 9J& 8J,)rQrRrSr\r}rrTrrrrerrrP __classcell__)ras@rOrXrXbsJ%"HU]]_   2D%% D6rVrXc Pt|j|j|j} gd} t | |j |j | |j} | j|i} d}|| d<|| d<||d| d<|jrt||}|| d<| | ||j|jd |jd <|jrii}t|jd z D];}|j!j#Dcic]\}}|d |z|c}}=j!|j#Dcic] \}}|d z| }}}j!|t%fdD}||fScc}}wcc}}w)N)r&r%r')rrr)r]r^r_r`r6rr)rrrr)rcrrrrqrc3>K|]}|vs||zywrZrU)rr loss_dict weight_dicts rOrz/LwDetrForObjectDetectionLoss..^s%T1 CSy|k!n,Ts )rr&r%r'rX num_labelsr_r6r"auxiliary_lossrclass_loss_coefficientbbox_loss_coefficientgiou_loss_coefficientr,decoder_layersrrrl)rrrrconfig outputs_class outputs_coordenc_outputs_classenc_outputs_coordkwargsr]r` criterion outputs_lossraux_weight_dictrIrr;enc_weight_dictrrrs @@rOLwDetrForObjectDetectionLossr-s%$$0@0@FL\L\G0F%%&&$$ ILLL#L!+L#'#L)-G,= (),/I$;;&JfJfgK%;;K  v,,q01 UA  " "{?P?P?R#Stq!A!A3KN#S T U?+1<1B1B1DEAq6z1}EOE' TiT TD - -- $TEs F F")NNNN)numpyr/rtorch.distributed distributedrtorch.nnrutilsrrloss_for_object_detectionrrr r r r r transformers.image_transformsrscipy.optimizerrModulerXrrUrVrOrsq ;F48t-8tvHbiiHb2.rV