37847c77889e39143034af9fe0d4b92bc74e992a,torch_geometric/datasets/utils/ply.py,,read_ply,#Any#,6

Before Change

    x = plydata["vertex"]["x"]
    y = plydata["vertex"]["y"]
    z = plydata["vertex"]["z"]
    vertices = np.stack((x, y, z), axis=1)

    faces = make2d(plydata["face"]["vertex_indices"])
    faces = faces.astype(np.int64)

    // Conver to torch tensors.
    vertices = torch.FloatTensor(vertices)
    faces = torch.LongTensor(faces)

    return vertices, faces

After Change

        plydata = PlyData.read(f)

    x = torch.FloatTensor(plydata["vertex"]["x"])
    y = torch.FloatTensor(plydata["vertex"]["y"])
    z = torch.FloatTensor(plydata["vertex"]["z"])
    pos = torch.stack([x, y, z], dim=1)
    input = torch.ones(pos.size(0))

    face = torch.LongTensor(make2d(plydata["face"]["vertex_indices"]))
    index = edges_from_faces(face)

    return Data(input, pos, index, None, None)

Instances