kaolin.datasets.shrec

class SHREC16(root: str, categories: list = ['alien'], mode: list = 'train')[source]

Class to help in loading the SHREC16 dataset.

SHREC16 is the dataset 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.

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).

  • mode (str, choices=['train', 'test']) – Whether to load the ‘train’ split or the ‘test’ split

Returns

Dictionary with keys: ‘vertices’ : vertices , ‘faces’ : faces

Return type

dict