o e)@sTddlZddlmZddZdddZdd Zdd d Zd dZddZ ddZ dS)N)PolygoncCsVt|tr t|}t|tjsJ|jddkr|jdks J|ddg}t|S)zConvert k points to 1 polygon. Args: points (ndarray or list): A ndarray or a list of shape (2k) that indicates k points. Returns: polygon (Polygon): A polygon object. r) isinstancelistnparrayZndarraysizeZreshaper)ZpointsZ point_matr ND:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/ppocr/utils/poly_nms.pypoints2polygons r -C6?cCsLt|tsJt|tsJ|dkr||@}n ||||@}|j|fS)aCalculate the intersection area between two polygon. Args: poly_det (Polygon): A polygon predicted by detector. poly_gt (Polygon): A gt polygon. Returns: intersection_area (float): The intersection area between two polygons. r)rrbufferarea)poly_detpoly_gtrZ poly_interr r r poly_intersection's    rcCsBt|tsJt|tsJ|j}|j}t||\}}|||S)zCalculate the union area between two polygon. Args: poly_det (Polygon): A polygon predicted by detector. poly_gt (Polygon): A gt polygon. Returns: union_area (float): The union area between two polygons. )rrrr)rrZarea_detZarea_gt area_inters_r r r poly_union;s   rTcCs@t|}|dkr dS|ddkr|sdS|ddkr|rdSdS)NrFrrT)len)xZ with_scorenumr r r valid_boundaryNsrcCs6t|dsJt|dsJt|}t|}t||S)zCalculate the IOU between two boundaries. Args: src (list): Source boundary. target (list): Target boundary. Returns: iou (float): The iou between two boundaries. F)rr poly_iou)srctargetZsrc_polyZ target_polyr r r boundary_iouZs   rcCsHt|tsJt|tsJt||\}}t||}|dkr dS||S)zCalculate the IOU between two polygons. Args: poly_det (Polygon): A polygon predicted by detector. poly_gt (Polygon): A gt polygon. Returns: iou (float): The IOU between two polygons. rg)rrrr)rrrrZ area_unionr r r rls  rc Cst|tsJtt|ddd}g}ddt|jdD}t|dkrv|||d ||ddd}t |d}t t|f}tt|D]}|||dd}t ||||<qOt ||k}t ||}t|dks&|S)NcSs|dS)Nrr )rr r r szpoly_nms..)keycSsg|]}|qSr r ).0ir r r szpoly_nms..rr)rrrr sortedrangeshaperappendtolistdeleteZzerosrwhere) Zpolygons thresholdZ keep_polyindexAZiou_listr#BZ remove_indexr r r poly_nmss      r0)r)T) numpyrZshapely.geometryrr rrrrrr0r r r r s