tfg.datasets.shapenet.Shapenet.builder_config_cls

A Shapenet config for loading the original .obj files.

tfg.datasets.shapenet.Shapenet.builder_config_cls(
    model_subpath='models/model_normalized.obj'
)

Attributes
name	Dataclass field
version	Dataclass field
release_notes	Dataclass field
supported_versions	Dataclass field
description	Dataclass field

Methods

from_dataset_info

@classmethod
from_dataset_info(
    info_proto: dataset_info_pb2.DatasetInfo
) -> Optional['BuilderConfig']

generate_examples

View source

generate_examples(
    base_dir, models
)

Yields examples.

The structure of the examples: { 'trimesh': tensorflow_graphics.datasets.features.TriangleMesh 'label': tensorflow_datasets.features.ClassLabel 'model_id': tensorflow_datasets.features.Text }

Args
base_dir	The base directory of shapenet.
models	The list of models in the split.

info

View source

info(
    dataset_builder
)

Delegated Shapenet._info.

split_generators

View source

split_generators(
    dl_manager, dataset_builder
)

Delegated Shapenet._split_generators.

Class Variables
description	None
release_notes	None
version	None