From 2672be5763b1555c5cdaccada08a3078f0d2e05f Mon Sep 17 00:00:00 2001 From: fangfangzk <75557988+fangfangzk@users.noreply.github.com> Date: Wed, 11 Dec 2024 21:56:34 +0800 Subject: [PATCH] fix:calcute the left_center_pt and right_center_pt from min_area_quad (#14363) --- ppocr/data/imaug/pg_process.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppocr/data/imaug/pg_process.py b/ppocr/data/imaug/pg_process.py index cdd816ccb1..d62aab6db8 100644 --- a/ppocr/data/imaug/pg_process.py +++ b/ppocr/data/imaug/pg_process.py @@ -265,8 +265,8 @@ def fit_and_gather_tcl_points_v2( ys, xs = np.where(tmp_image > 0) xy_text = np.array(list(zip(xs, ys)), dtype="float32") - left_center_pt = ((min_area_quad[0] - min_area_quad[1]) / 2.0).reshape(1, 2) - right_center_pt = ((min_area_quad[1] - min_area_quad[2]) / 2.0).reshape(1, 2) + left_center_pt = ((min_area_quad[0] + min_area_quad[3]) / 2.0).reshape(1, 2) + right_center_pt = ((min_area_quad[1] + min_area_quad[2]) / 2.0).reshape(1, 2) proj_unit_vec = (right_center_pt - left_center_pt) / ( np.linalg.norm(right_center_pt - left_center_pt) + 1e-6 )