akko/diffvg
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Convert arc points to torch.float

Browse Source
This commit is contained in:
Oren Katzir
2023-01-09 09:25:35 +02:00
committed by Akko
parent 28e9dc0809
commit 0d69c2b05e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pydiffvg/shape.py
View File

@@ -164,7 +164,7 @@ def from_svg_path(path_str, shape_to_canvas = torch.eye(3), force_close = False)
assert(e.end.imag == points[0][1]) assert(e.end.imag == points[0][1])
else: else:
points.append((e.end.real, e.end.imag)) points.append((e.end.real, e.end.imag))
points = torch.tensor(points) points = torch.tensor(points, dtype=torch.float)
points = torch.cat((points, torch.ones([points.shape[0], 1])), dim = 1) @ torch.transpose(shape_to_canvas, 0, 1) points = torch.cat((points, torch.ones([points.shape[0], 1])), dim = 1) @ torch.transpose(shape_to_canvas, 0, 1)
points = points / points[:, 2:3] points = points / points[:, 2:3]
points = points[:, :2].contiguous() points = points[:, :2].contiguous()