From d5f329d04566af84ff3734d8ef66f78f2b88fe89 Mon Sep 17 00:00:00 2001 From: yfzhou Date: Thu, 12 May 2022 12:01:39 +0800 Subject: [PATCH] the file has no function [self.vector_slope] MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit self.vector_slope 方法不存在，应该是漏掉了 --- ppocr/data/imaug/fce_targets.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/ppocr/data/imaug/fce_targets.py b/ppocr/data/imaug/fce_targets.py index 18184808..4d1903c0 100644 --- a/ppocr/data/imaug/fce_targets.py +++ b/ppocr/data/imaug/fce_targets.py @@ -22,6 +22,9 @@ from numpy.fft import fft from numpy.linalg import norm import sys +def vector_slope(vec): + assert len(vec) == 2 + return abs(vec[1] / (vec[0] + 1e-8)) class FCENetTargets: """Generate the ground truth targets of FCENet: Fourier Contour Embedding @@ -233,9 +236,9 @@ class FCENetTargets: head_inds = [head_start, head_end] tail_inds = [tail_start, tail_end] else: - if self.vector_slope(points[1] - points[0]) + self.vector_slope( - points[3] - points[2]) < self.vector_slope(points[ - 2] - points[1]) + self.vector_slope(points[0] - points[ + if vector_slope(points[1] - points[0]) + vector_slope( + points[3] - points[2]) < vector_slope(points[ + 2] - points[1]) + vector_slope(points[0] - points[ 3]): horizontal_edge_inds = [[0, 1], [2, 3]] vertical_edge_inds = [[3, 0], [1, 2]] -- GitLab