Skip to content

Commit

Permalink
Add additional examples for graphs, meshes and point clouds (#337)
Browse files Browse the repository at this point in the history
Summary:
This PR adds additional examples of `DataPipes` for loading graphs, meshes and point clouds as part of my recent efforts to bring [`torchdata` support to PyTorch Geometric](https://github.com/pyg-team/pytorch_geometric/blob/master/examples/datapipe.py).

Pull Request resolved: #337

Reviewed By: ejguan

Differential Revision: D35228267

Pulled By: NivekT

fbshipit-source-id: c83f75e88c612a5e5cd972eba4f786f5a8a64058
  • Loading branch information
rusty1s authored and facebook-github-bot committed Mar 30, 2022
1 parent 99dc849 commit 71a16a3
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions docs/source/examples.rst
Original file line number Diff line number Diff line change
Expand Up @@ -84,3 +84,22 @@ The `Criteo dataset <https://ailab.criteo.com/download-criteo-1tb-click-logs-dat
and click feedback for millions of display advertisements. It aims to benchmark algorithms for
click through rate (CTR) prediction. You can find a prototype stage implementation of the
`dataset with DataPipes in TorchRec <https://github.com/pytorch/torchrec/blob/main/torchrec/datasets/criteo.py>`_.

Graphs, Meshes and Point Clouds
-------------------------------

MoleculeNet
^^^^^^^^^^^
`MoleculeNet <https://moleculenet.org/>`_ is a benchmark specially designed for testing machine learning methods of
molecular properties. You can find an implementation of the
`HIV dataset with DataPipes in PyTorch Geometric <https://github.com/pyg-team/pytorch_geometric/blob/master/examples/datapipe.py>`_,
which includes converting SMILES strings into molecular graph representations.

Princeton ModelNet
^^^^^^^^^^^^^^^^^^

The Princeton ModelNet project provides a comprehensive and clean collection of 3D CAD models across various object types.
You can find an implementation of the
`ModelNet10 dataset with DataPipes in PyTorch Geometric <https://github.com/pyg-team/pytorch_geometric/blob/master/examples/datapipe.py>`_,
which includes reading in meshes via `meshio <https://github.com/nschloe/meshio>`_, and sampling of points from object surfaces and dynamic
graph generation via `PyG's functional transformations <https://pytorch-geometric.readthedocs.io/en/latest/modules/transforms.html>`_.

0 comments on commit 71a16a3

Please sign in to comment.