Unverified Commit e4f998b9 authored by zhoujun's avatar zhoujun Committed by GitHub
Browse files

Merge pull request #6261 from Bourne-M/patch-5

the file has no  function [self.vector_slope]
parents 791712e2 d5f329d0
......@@ -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]]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment