Generators reference#

class blenderline.generators.ImageDatasetGenerator#

Generator for image datasets.

__init__(name, target, scene_manager, hdr_manager, background_manager, item_manager)#

Create dataset generator.

Parameters:

  • name (str) – name of dataset. Output folder will be target/<name>/<split>.

  • target (pathlib.Path) – base directory in which to put data.

  • scene_manager (SceneManager) – scene manager.

  • hdr_manager (HDRManager) – HDR background manager.

  • background_manager (BackgroundManager) – background manager.

  • item_manager (ItemManager) – item manager.

Return type:

None

register_label(label, label_name)#

Register dataset label to generate

Parameters:

  • label (int) – label index found in mask filenames.

  • label_name (str) – semantic name for label index.

Return type:

None

register_split(name, size)#

Register dataset split to generate

Parameters:

  • name (str) – name of split to generate within dataset name output folder.

  • size (int) – number of images to generate.

Return type:

None