kaolin.io.shrec

API

class kaolin.io.shrec.SHREC16(root: str, categories: Optional[list] = None, split: str = 'train')

Bases: Generic[torch.utils.data.dataset.T_co]

Dataset class for SHREC16, used for the “Large-scale 3D shape retrieval from ShapeNet Core55” contest at Eurographics 2016. More details about the challenge and the dataset are available here.

The __getitem__ method will return a KaolinDatasetItem, with its data field containing a kaolin.io.obj.return_type.

Parameters

  • root (str) – Path to the root directory of the dataset.

  • categories (list) – List of categories to load (each class is specified as a string, and must be a valid SHREC16 category). If this argument is not specified, all categories are loaded by default.

  • split (str) – String to indicate what split to load, among [“train”, “val”, “test”]. Default: “train”.

get_attributes(index)

Returns the attributes at the given index. Attributes are usually not transformed by wrappers such as ProcessedDataset.

get_cache_key(index)
get_data(index)

Returns the data at the given index.