torch_geometric.datasets.PCPNetDataset

class PCPNetDataset(root: str, category: str, split: str = 'train', transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, pre_filter: Optional[Callable] = None, force_reload: bool = False)[source]

Bases: InMemoryDataset

来自“PCPNet: Learning Local Shape Properties from Raw Point Clouds”论文的PCPNet数据集，包含30个形状，每个形状以点云形式给出，密集采样了10万个点。对于每个形状，表面法线和局部曲率作为节点特征给出。

Parameters:

  • root (str) – Root directory where the dataset should be saved.

  • category (str) – 训练集的类别（对于 split="train" 或 split="val"，可以是 "NoNoise", "Noisy", "VarDensity", "NoisyAndVarDensity"， 或者对于 split="test"，可以是 "All", "LowNoise", "MedNoise", "HighNoise", :obj:”VarDensityStriped”, "VarDensityGradient"）。

  • split (str, optional) – 如果为 "train"，则加载训练数据集。 如果为 "val"，则加载验证数据集。 如果为 "test"，则加载测试数据集。（默认值："train"）

  • transform (callable, optional) – 一个函数/转换，它接收一个 torch_geometric.data.Data 对象并返回一个转换后的 版本。数据对象将在每次访问之前进行转换。 (默认: None)

  • pre_transform (callable, optional) – 一个函数/转换，接收一个torch_geometric.data.Data对象并返回一个转换后的版本。数据对象将在保存到磁盘之前进行转换。（默认值：None）

  • pre_filter (callable, optional) – 一个函数，接收一个 torch_geometric.data.Data 对象并返回一个布尔值，表示该数据对象是否应包含在最终数据集中。（默认值：None）

  • force_reload (bool, optional) – Whether to re-process the dataset. (default: False)