U ~c)@sTddlZddlmZddZdddZdd Zdd d Zd dZddZ ddZ dS)N)PolygoncCsVt|trt|}t|tjs$t|jddkr<|jdks@t|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) isinstancelistnparrayndarrayAssertionErrorsizereshaper)pointsZ point_matrX/Users/zhoujun20/Desktop/工作相关/一致性/训练/PaddleOCR/ppocr/utils/poly_nms.pypoints2polygons  r-C6?cCsLt|tstt|tst|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)rrr bufferarea)poly_detpoly_gtrZ poly_interrrrpoly_intersection's  rcCsBt|tstt|tst|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. )rrr rr)rrZarea_detZarea_gt area_inters_rrr poly_union;s rTcCs@t|}|dkrdS|ddkr(|s(dS|ddkr<|rzpoly_nms..)keycSsg|]}|qSrr).0irrr szpoly_nms..rr)rrr rr sortedrangeshaperappendtolistdeletezerosr#where) polygons thresholdZ keep_polyindexAZiou_listr(BZ remove_indexrrrpoly_nmss  r7)r)T) numpyrshapely.geometryrrrrrr#r r7rrrrs