the file has no function [self.vector_slope]
self.vector_slope 方法不存在,应该是漏掉了
This commit is contained in:
parent
400989320b
commit
d5f329d045
|
@ -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]]
|
||||
|
|
Loading…
Reference in New Issue