o *j3@sddlZddlZddlZddlmZdddZdddZdd d Zd d Z d dZ      dddZ ddZ dddZ dddZdS) N)nmscCsFtjt|t|gdd\}}t||fddd||dfS)NZij)Zindexing)torchZmeshgridZarangestackviewfloat)nxnyZyvZxvr t/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/image_driving_perception/utils.py _make_grid s $rc Csg}gd}tdD]j}||j\}}}}|||dd||ddddd||<||} t||||j} | dddfd d | ||| dddf<| dddfdd||| dddf<| | |d dq t |d}|S) N) Urrr.g@g?) rangeshaperZpermute contiguousZsigmoidrtodeviceappendrcat) predZ anchor_gridzstibs_r r ygrr r r split_for_trace_models   0, r%iicCs|dur.t|d|d|d|d}|d|d|d|d|d|df}n |dd}|d}|ddddgf|d8<|ddddgf|d8<|ddddf|<t|||S)Nrrrrr)min clip_coords)Z img1_shapeZcoordsZ img0_shapeZ ratio_padZgainpadr r r scale_coords s    r*cCst|dddfd|d|dddfd|d|dddfd|d|dddfd|ddS)Nrrrr)Zclamp_)boxesZ img_shaper r r r(2s r(cCst|tjr |nt|}|dddf|dddfd|dddf<|dddf|dddfd|dddf<|dddf|dddfd|dddf<|dddf|dddfd|dddf<|S)Nrrrr) isinstancerZTensorclonenpcopy)xr#r r r xywh2xyxy:s 0000r1333333??Fr cCs|jdd}|d|k}d} d} d} d} d} ||d kM}d }t}tjd |jd g|jd }t|D]\}}|||}|rt||r||}tjt||df|jd }|ddd df|ddddf<d|dddf<d|tt||ddd fdf<t ||fd }|jd sq5|ddddf|ddddf9<t |ddddf}|r|ddddf|kj d dj \}}t |||||ddf|dddf fd }n#|ddddfjd dd\}}t ||| fd |d|k}|dur'||ddddftj||jd kd }|jd }|s0q5|| krG||dddfjddd| }|ddddf|rUd n| }|ddddf||dddf}}t|||}|jd | kr|d| }|rd |kr|dkrt||||k}||d}t||ddddf |jd dd||ddf<| r||d d k}||||<t|| krtd| d|Sq5|S)zRuns Non-Maximum Suppression (NMS) on inference results Returns: list of detections, on (n,6) tensor per image [xyxy, conf, cls] r).rii,i0ug$@TrF)r)rrNrg?)as_tuple)Zkeepdimrr5)Z descendinggp@zWARNING: NMS time limit z s exceeded)rtimerZzerosr enumeratelenrlongrr1ZnonzeroTr maxrZtensoranyZargsortrbox_ioummsumprint)Z predictionZ conf_thresZ iou_thresclassesZagnosticZ multi_labellabelsncZxcZmax_whZmax_detZmax_nmsZ time_limitZ redundantmergetoutputxir0Zlbsvboxr jZconfncr+ZscoresZiouweightsr r r non_max_suppressionDs   $( ,&6"  .  "   " rOcCsdd}||j}||j}t|dddddf|ddddft|dddddf|ddddfdd}||dddf||S)aX Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in (x1, y1, x2, y2) format. Args: box1 (Tensor[N, 4]) box2 (Tensor[M, 4]) Returns: iou (Tensor[N, M]): the NxM matrix containing the pairwise IoU values for every element in boxes1 and boxes2 cSs |d|d|d|dS)Nrrrrr )rJr r r box_areas zbox_iou..box_areaNrr)r;rr'r<clampprod)Zbox1Zbox2rPZarea1Zarea2interr r r r>s  ,, r>cCs<tjjj||dd}t|d\}}|}|SNZbilinear)sizemoder) rnn functional interpolater<intsqueezecpunumpy)Z da_predict out_shapeZ da_seg_maskr"r r r driving_area_mask r_cCs<tjjj||dd}t|d}|}|SrT) rrWrXrYroundr[rZr\r])Z ll_predictr^Z ll_seg_maskr r r lane_line_maskr`rb)rr)NN)r&N)r2r3NFFr )Nr&)r7r]r.rZtorchvision.opsrrr%r*r(r1rOr>r_rbr r r r s&      `