o ߥi@sTddlmZddlmZddlZddlZGdddeZddZ dd Z d d Z dS) )product)ceilNcs&eZdZdfdd ZddZZS)PriorBoxNtraincsRtt|d_|d_|d_|_fddjD_d_dS)N min_sizesstepsclipcs0g|]}tjd|tjd|gqS)r)r image_size).0stepselfh/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/cv/face_detection/retinaface/utils.py s z%PriorBox.__init__..s) superr__init__rrrr feature_mapsname)rcfgr phase __class__r rrs     zPriorBox.__init__c sg}tjD]Z\}j}tt|dt|dD]C\}}|D]<}|jd}|jd}fdd|dfD} fdd|dfD} t| | D] \} } || | ||g7}qRq#qqt|dd} j ru| j ddd | S) Nrr c$g|]}|jjdqS)r rr )r xkrrrr"z$PriorBox.forward..g?cr)rr)r yrrrr&r )maxmin) enumeraterrrranger torchTensorviewrclamp_)ranchorsfrijmin_sizes_kxs_kydense_cxdense_cycycxoutputrrrforwards, "  zPriorBox.forward)Nr)__name__ __module__ __qualname__rr8 __classcell__rrrrr s rcCs|dddf}|dddf}|dddf}|dddf}|dddf}||d||d}|ddd}g} |jdkr|d} | | t|| ||dd} t|| ||dd} t|| ||dd} t|| ||dd}td| | d}td|| d}||}||| ||dd|}t||kd}||d}|jdksD| S) zPure Python NMS baseline.Nrr r#r"g)argsortsizeappendnpmaximumminimumwhere)detsthreshx1y1x2y2scoresareasorderkeepr.xx1yy1xx2yy2whinterovrindsrrr py_cpu_nms4s.     rYc Cst|ddddf|ddddf|d|ddddf|ddddft|ddddf|dfd}|ddddf|ddddfd8<|ddddf|ddddf7<|S)aDecode locations from predictions using priors to undo the encoding we did for offset regression at train time. Args: loc (tensor): location predictions for loc layers, Shape: [num_priors,4] priors (tensor): Prior boxes in center-offset form. Shape: [num_priors,4]. variances: (list[float]) Variances of priorboxes Return: decoded bounding box predictions Nr=rr )r(catexp)locpriors variancesboxesrrrdecodeTs B40,r`c Csp|ddddf|ddddf|d|ddddf}|ddddf|ddddf|d|ddddf}|ddddf|ddddf|d|ddddf}|ddddf|ddddf|d|ddddf}|ddddf|ddddf|d|ddddf}tj|||||fdd }|S) aDecode landm from predictions using priors to undo the encoding we did for offset regression at train time. Args: pre (tensor): landm predictions for loc layers, Shape: [num_priors,10] priors (tensor): Prior boxes in center-offset form. Shape: [num_priors,4]. variances: (list[float]) Variances of priorboxes Return: decoded landm predictions Nr=rr# r )dim)r(rZ) prer]r^abcdelandmsrrr decode_landmisD DDDDrl) itertoolsrmathrnumpyrBr(objectrrYr`rlrrrrs  (