Functions — github.com/barakooda/3d_scene

hub / github.com/barakooda/3d_scene_segmentation / functions

Functions76 in github.com/barakooda/3d_scene_segmentation

⨍ Functions 76 ◇ Types & classes 1

↓ 4 callersFunctionsetup_logger

(name, level=logging.INFO, log_format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')

src/logger_config.py:3

↓ 3 callersFunctionstring_to_rgba_color

(name: str, color_bit_depth: int = 16, channels_num:int=3,

src/seg_utils.py:80

↓ 2 callersFunctionadd_material_to_object

(obj: bpy.types.Object, material: bpy.types.Material)

src/utils.py:53

↓ 2 callersFunctionclean_all_material_nodes

(material_nodes: bpy.types.bpy_prop_collection)

src/utils.py:22

↓ 2 callersFunctioncreate_segmentation_material

Create a segmentation material with the given name, attribute name, and attribute type. :param name: Name of the material. :param at

src/seg_materials.py:8

↓ 2 callersFunctiondelete_camera_and_bouding_box

()

src/object_classifier_utils.py:20

↓ 2 callersMethodencode_image

(self, image_tensor)

src/ai_image_classifier.py:135

↓ 2 callersFunctionremove_prefix_from_string

(group_name:str ,prefix_list:list)

src/utils.py:43

↓ 2 callersFunctionset_segmentation_render_settings

(scene_name: str = "Scene")

src/render_settings.py:7

↓ 1 callersMethod_generate_labels

(self)

src/ai_image_classifier.py:60

↓ 1 callersMethod_load_model

(self)

src/ai_image_classifier.py:50

↓ 1 callersFunctionactivate_lights_in_render

(lights_list)

src/utils.py:150

↓ 1 callersFunctionadd_color_attribute_to_face_corner_of_vertex_group

( obj: bpy.types.Object = None, segmentation_dic: dict = None, attribute_name: str = ATTRIBUTE_NAM

src/seg_utils.py:347

↓ 1 callersMethodclassify_images

Classifies a list of OpenCV images (NumPy arrays) and returns the most frequent label.

src/ai_image_classifier.py:179

↓ 1 callersFunctionclassify_objects

(classifier, globally_visible_object_names)

src/object_classifier_utils.py:444

↓ 1 callersFunctionconvert_all_curve_particle_instances_to_mesh

()

src/seg_utils.py:17

↓ 1 callersFunctionconvert_curve_to_mesh

Converts a Blender curve object to a mesh object if it is not already a mesh. Args: obj_name (str): The name of the object to conver

src/seg_utils.py:31

↓ 1 callersFunctioncreate_bbox_and_cameras

(obj: bpy.types.Object)

src/object_classifier_utils.py:67

↓ 1 callersFunctioncreate_colors_by_groups_names

( obj: bpy.types.Object, color_bit_depth: int = 16, prefix_list: list = None)

src/seg_utils.py:194

↓ 1 callersFunctioncreate_hair_particle_system_material

Create a hair particle system material for the given particle system and color. :param particle_system: The particle system to create th

src/seg_materials.py:72

↓ 1 callersFunctioncreate_new_world_material_with_sky

(scene: bpy.types.Scene)

src/object_classifier_utils.py:26

↓ 1 callersFunctioncreate_shader_node

( material: bpy.types.Material, node_name: str = "attribute_node_name01", node_type: str = "Shader

src/utils.py:26

↓ 1 callersFunctionencode_image

Encodes a single image using the model.

src/object_classifier_utils.py:339

↓ 1 callersFunctionfilter_by_startwith_names

(objects: list = None, prefixes: list = None)

src/utils.py:75

↓ 1 callersFunctiongenerate_noun_labels

(word_list, max_labels=10000)

src/object_classifier_utils.py:277

↓ 1 callersMethodgenerate_noun_labels

(self, word_list, max_labels=10000)

src/ai_image_classifier.py:64

↓ 1 callersFunctionget_active_lights

(scene: bpy.types.Scene)

src/utils.py:136

↓ 1 callersFunctionget_all_particle_systems_instance_objects

()

src/seg_utils.py:69

↓ 1 callersFunctionget_best_label

Finds the label with the maximum count.

src/object_classifier_utils.py:375

↓ 1 callersMethodget_best_label

(self)

src/ai_image_classifier.py:165

↓ 1 callersFunctionget_bytes_array_from_string

(input_string: str,split_size_in_bytes:int = 2,channels_num:int = 3 )

src/seg_utils.py:98

↓ 1 callersFunctionget_color_from_split_bytes_list

(key_list: list, alpha: int = MAX_UINT16)

src/seg_utils.py:119

↓ 1 callersFunctionget_data_per_polygon

(obj: bpy.types.Object = None)

src/seg_utils.py:238

↓ 1 callersFunctionget_globally_visible_object_names

(scene_name : str = "Scene",view_layer_name : str = "ViewLayer")

src/utils.py:155

↓ 1 callersFunctionget_objects_name_to_object_attribute

(globally_visible_object_names:list,attribute_name:str)

src/utils.py:189

↓ 1 callersFunctionget_polygon_indices_for_vertex_group

(obj, segmentation_dic)

src/seg_utils.py:302

↓ 1 callersFunctionget_string_without_trailing_digits

(string:str)

src/utils.py:201

↓ 1 callersFunctionget_unique_file_path

Generates a unique file path by appending a random UUID string to the file name. Args: - file_path: The initial path of the file.

src/utils.py:302

↓ 1 callersFunctionget_vertices_data_per_group

Collects vertices data for specified vertex groups within a Blender object. Parameters: obj (bpy.types.Object): The Blender object with

src/seg_utils.py:251

↓ 1 callersFunctionhide_all_except_one_globally

(visible_object_names:list[str], chosen_object_name:str)

src/utils.py:163

↓ 1 callersFunctionis_contain_vertex_groups

(obj: bpy.types.Object, vertex_group_filter: str = None)

src/seg_utils.py:153

↓ 1 callersFunctionis_point_in_bounding_box

(polygon_data: dict, vertex_group_data: dict)

src/utils.py:92

↓ 1 callersFunctionload_and_preprocess_image

Loads and preprocesses a single image.

src/object_classifier_utils.py:333

↓ 1 callersFunctionmap_polygons_to_vertex_groups

(data_per_polygon, data_per_vertex_group)

src/seg_utils.py:220

↓ 1 callersMethodpreprocess_image

Preprocesses an image array using torchvision's functional API without PIL.

src/ai_image_classifier.py:105

↓ 1 callersMethodprocess_image

(self, image_array)

src/ai_image_classifier.py:140

↓ 1 callersFunctionprocess_image_file

Processes a single image file and returns the best label.

src/object_classifier_utils.py:345

↓ 1 callersFunctionprocess_in_batches

(image_features, text_inputs, labels, model, batch_size=500)

src/object_classifier_utils.py:308

↓ 1 callersMethodprocess_in_batches

(self, image_features)

src/ai_image_classifier.py:79

↓ 1 callersFunctionrender_from_cameras_to_memory

Renders images from a list of cameras in a specified scene and returns them as NumPy arrays using OpenCV.

src/object_classifier_utils.py:192

↓ 1 callersFunctionsave_dict_to_json

(file_path:str,dictionary:dict)

src/utils.py:184

↓ 1 callersFunctionset_cameras_for_segmentation

()

src/render_settings.py:27

↓ 1 callersFunctionset_faces_corners_color_by_group_name

Sets colors for mesh vertex corners based on vertex group names. Parameters: obj (bpy.types.Object): The mesh object. group_colors (

src/seg_utils.py:315

↓ 1 callersFunctionset_object_segmentation

(obj:bpy.types.Object = None,semantic_name:str = UNDEFINED_OBJECT,object_segmentation:str = OBJECT_SEGMENTATI

src/seg_utils.py:128

↓ 1 callersFunctionset_particle_system_segmentation

(obj :bpy.types.Object = None,segmentation_custom_property:str = None)

src/seg_utils.py:361

↓ 1 callersFunctionset_seed

(seed,device = None, cuda_deterministic=True)

src/object_classifier_utils.py:389

↓ 1 callersMethodset_seed

(self, seed, cuda_deterministic=True)

src/ai_image_classifier.py:30

↓ 1 callersFunctionset_sub_segmentation

(obj:bpy.types.Object = None,prefix_list:list = None,sub_segmentation:str = SUB_SEGMENTATION, sub_segmentatio

src/seg_utils.py:166

↓ 1 callersFunctionsplit_bytes_array

(bytes_array: bytes, split_size: int = 2)

src/seg_utils.py:113

↓ 1 callersFunctiontokenize_labels

(labels, device)

src/object_classifier_utils.py:296

↓ 1 callersMethodtokenize_labels

(self, labels, device)

src/ai_image_classifier.py:76

↓ 1 callersFunctionunhide_globally_objects

(visible_object_names :list[str])

src/utils.py:175

↓ 1 callersFunctionupdate_final_labels

Updates the final label counts.

src/object_classifier_utils.py:366

↓ 1 callersMethodupdate_final_labels

(self, best_label)

src/ai_image_classifier.py:158

Method__init__

(self, model_accuracy='s', max_labels=10000,seed=None)

src/ai_image_classifier.py:14

Functionapply_rotation_to_vertices

(obj: bpy.types.Object)

src/utils.py:109

Functioncreate_stitched_image

Creates a 2x3 stitched RGBA image from 6 rendered images and places it on the top-right corner of an empty image. Args: - rendered_i

src/utils.py:241

Functiondeactivate_lights_in_render

(lights_list)

src/utils.py:146

Functionextract_nouns

(text)

src/object_classifier_utils.py:482

Functiongenerate_label_from_images

(model_accuracy : str = 's',max_labels:int=10000)

src/object_classifier_utils.py:403

Functionget_context_window

src/object_classifier_utils.py:436

Functionprioritize_seed_words

(nouns, seed_words)

src/object_classifier_utils.py:489

Functionrender_viewport_image

(file_path:str,scene_name:str = "Scene",view_layer_name:str = "ViewLayer",camera_name:str = "Camera")

src/utils.py:213

Functionsave_labels_to_file

(labels, file_path)

src/object_classifier_utils.py:300

Functionvisualize_images

(rendered_images)

src/object_classifier_utils.py:257