From 3728183276617ac5f8e27fbabfc5862cb1d5694f Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Sun, 7 Feb 2021 08:09:39 +0000 Subject: [PATCH] fix are calc for polygon --- ppocr/data/imaug/make_shrink_map.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/ppocr/data/imaug/make_shrink_map.py b/ppocr/data/imaug/make_shrink_map.py index a66706f2..8e612235 100644 --- a/ppocr/data/imaug/make_shrink_map.py +++ b/ppocr/data/imaug/make_shrink_map.py @@ -84,11 +84,12 @@ class MakeShrinkMap(object): return polygons, ignore_tags def polygon_area(self, polygon): - # return cv2.contourArea(polygon.astype(np.float32)) - edge = 0 - for i in range(polygon.shape[0]): - next_index = (i + 1) % polygon.shape[0] - edge += (polygon[next_index, 0] - polygon[i, 0]) * ( - polygon[next_index, 1] - polygon[i, 1]) - - return edge / 2. + """ + compute polygon area + """ + area = 0 + q = polygon[-1] + for p in polygon: + area += p[0] * q[1] - p[1] * q[0] + q = p + return area / 2.0 -- GitLab