palletdatagenerator.generator module

PalletDataGenerator - Main Generator Class

Uses modular architecture with separate mode classes for single_pallet and warehouse modes. Each mode implements the exact logic from the original generator files.

palletdatagenerator.generator.ensure_dependencies()

Ensure required dependencies are installed.

class palletdatagenerator.generator.PalletDataGenerator(mode: str = 'single_pallet', scene_path: str | None = None)

Bases: object

Main generator that delegates to appropriate mode classes. Replicates the exact functionality of original files using modular architecture.

__init__(mode: str = 'single_pallet', scene_path: str | None = None)

generate(scene_path: Path | None = None, num_frames: int = 50, output_dir: Path | None = None, resolution: tuple[int, int] | None = None) → dict[str, Any]

Generate dataset using the appropriate mode class.

palletdatagenerator.generator.enable_gpu(preferred=None)

Enable GPU rendering with a platform-aware preference order.

palletdatagenerator.generator.ensure(path)

palletdatagenerator.generator.build_folders(root)

palletdatagenerator.generator.configure_render(cfg)

palletdatagenerator.generator.setup_compositor_nodes(paths, _cfg)

palletdatagenerator.generator.draw_3d_bbox_edges(draw, corners_2d, color, width=2)

palletdatagenerator.generator.project_points_accurate(points, cam, sc)

palletdatagenerator.generator.create_analysis_image_multi(rgb_path, bboxes2d, bboxes3d, all_pockets_world, cam_obj, sc, output_path, frame_id)

EXACT analysis image creation from original.

palletdatagenerator.generator.main_single_pallet(CONFIG)

EXACT main function from original one_pallet_generator.py

palletdatagenerator.generator.main_warehouse(CONFIG)

EXACT main function from original warehouse_generator.py