Source code for torch_geometric.transforms.delaunay

import torch
import scipy.spatial

[docs]class Delaunay(object): r"""Computes the delaunay triangulation of a set of points.""" def __call__(self, data): if data.pos.size(0) > 3: pos = data.pos.cpu().numpy() tri = scipy.spatial.Delaunay(pos, qhull_options='QJ') face = torch.from_numpy(tri.simplices) elif data.pos.size(0) == 3: face = torch.tensor([[0, 1, 2]]) else: raise ValueError( 'Not enough points to contruct Delaunay triangulation, got {} ' 'but expected at least 3'.format(data.pos.size(0))) data.face = face.t().contiguous().to(data.pos.device, torch.long) return data def __repr__(self): return '{}()'.format(self.__class__.__name__)