palletdatagenerator.modes package

Submodules

• palletdatagenerator.modes.base_generator module
  • BaseGenerator
    • BaseGenerator.__init__()
    • BaseGenerator.setup_folders()
    • BaseGenerator.configure_render()
    • BaseGenerator.setup_compositor_nodes()
    • BaseGenerator.setup_environment()
    • BaseGenerator.setup_random_background()
    • BaseGenerator.create_floor_plane()
    • BaseGenerator.randomize_object_material()
    • BaseGenerator.draw_3d_bbox_edges()
    • BaseGenerator.project_points_accurate()
    • BaseGenerator.create_analysis_image_multi()
    • BaseGenerator.create_random_lights()
    • BaseGenerator.get_bbox_2d_accurate()
    • BaseGenerator.bbox_3d_oriented()
    • BaseGenerator.project_points()
    • BaseGenerator.hole_bboxes_3d()
    • BaseGenerator.auto_expose_frame()
    • BaseGenerator.detect_faces_in_scene()
    • BaseGenerator.get_all_faces_from_bbox()
    • BaseGenerator.filter_side_faces()
    • BaseGenerator.create_3d_debug_visualization()
    • BaseGenerator.create_3d_debug_visualization_with_faces()
    • BaseGenerator.select_faces_by_camera_proximity()
    • BaseGenerator.check_faces_adjacent()
    • BaseGenerator.check_2d_middle_view()
    • BaseGenerator.check_geometric_adjacency_only()
    • BaseGenerator.check_2d_surface_quality()
    • BaseGenerator.check_face_behind_primary()
    • BaseGenerator.detect_overlapping_keypoints()
    • BaseGenerator.draw_overlapping_keypoint_circles()
    • BaseGenerator.draw_single_keypoint_labels()
    • BaseGenerator.draw_overlapping_keypoint_labels()
    • BaseGenerator.calculate_face_normal()
    • BaseGenerator.generate_keypoints_for_frame()
    • BaseGenerator.generate_face_2d_boxes()
    • BaseGenerator.generate_face_3d_coordinates()
    • BaseGenerator.create_interactive_3d_figure()
    • BaseGenerator.generate_face_keypoints()
    • BaseGenerator.check_keypoint_visibility()
    • BaseGenerator.save_keypoints_labels()
• palletdatagenerator.modes.single_pallet module
  • SinglePalletMode
    • SinglePalletMode.__init__()
    • SinglePalletMode.position_camera_for_side_face()
    • SinglePalletMode.generate_frames()
    • SinglePalletMode.apply_pallet_movement()
    • SinglePalletMode.prepare_pallets()
    • SinglePalletMode.duplicate_pallets_if_needed()
    • SinglePalletMode.position_pallets()
    • SinglePalletMode.setup_lighting()
    • SinglePalletMode.create_random_lights()
    • SinglePalletMode.handle_attached_boxes()
    • SinglePalletMode.ATTACHED_GROUP_PREFIX
    • SinglePalletMode.get_detections()
    • SinglePalletMode.auto_expose_frame()
    • SinglePalletMode.save_frame()
    • SinglePalletMode.project_holes_and_write_labels()
    • SinglePalletMode.save_final_outputs()
    • SinglePalletMode.apply_initial_transform()
    • SinglePalletMode.save_generated_scene()
• palletdatagenerator.modes.warehouse module
  • WarehouseMode
    • WarehouseMode.__init__()
    • WarehouseMode.generate_frames()
    • WarehouseMode.find_warehouse_objects()
    • WarehouseMode.generate_warehouse_path()
    • WarehouseMode.find_nearest_pallet()
    • WarehouseMode.position_camera_on_path()
    • WarehouseMode.randomize_scene_objects()
    • WarehouseMode.generate_pallet_box_group()
    • WarehouseMode.cleanup_generated_boxes()
    • WarehouseMode.find_pallet_box_relationships()
    • WarehouseMode.get_visible_pallets()
    • WarehouseMode.randomize_lighting()
    • WarehouseMode.save_warehouse_frame_outputs()
    • WarehouseMode.write_warehouse_annotations()
    • WarehouseMode.restore_scene_objects()
    • WarehouseMode.save_generated_scene()

Module contents

Modes package - Contains generator mode implementations

Each mode implements the exact logic from the original generator files.