def __init__(self, config): Module.__init__(self, config) self.randomization_level = self.config.get_float("randomization_level", 0.2) self.randomize_textures_only = self.config.get_float("randomize_textures_only", False) self.output_textures_only = self.config.get_float("output_textures_only", True) self.scene_materials = []
def __init__(self, config): Module.__init__(self, config) self.source_frame = self.config.get_list("source_frame", ["X", "Y", "Z"]) self.cam_pose_collection = ItemCollection( self._add_cam_pose, self.config.get_raw_dict("default_cam_param", {}))
def __init__(self, config): Module.__init__(self, config) self._folder_path = "" self._used_assets = [] self._add_cp = {} self._preload = False self._fill_used_empty_materials = False
def __init__(self, config: Config): Module.__init__(self, config) self.emission_strength = self.config.get_float("emission_strength", 10.0) self.keep_using_base_color = self.config.get_bool( "keep_using_base_color", False) self.emission_color = self.config.get_vector3d("emission_color", None)
def __init__(self, config: Config): """ This function is called by the Pipeline object, it initialized the object and reads all important config values :param config: The config object used for this module, specified by the .yaml file """ Module.__init__(self, config) self.used_floor_area = self.config.get_float("floor_area") self.amount_of_extrusions = self.config.get_int( "amount_of_extrusions", 0) self.fac_from_square_room = self.config.get_float( "fac_base_from_square_room", 0.3) self.corridor_width = self.config.get_float("minimum_corridor_width", 0.9) self.wall_height = self.config.get_float("wall_height", 2.5) self.amount_of_floor_cuts = self.config.get_int( "amount_of_floor_cuts", 2) self.only_use_big_edges = self.config.get_bool("only_use_big_edges", True) self.create_ceiling = self.config.get_bool("create_ceiling", True) self.assign_material_to_ceiling = self.config.get_bool( "assign_material_to_ceiling", False) self.tries_per_face = self.config.get_int("placement_tries_per_face", 3) self.amount_of_objects_per_sq_meter = self.config.get_float( "amount_of_objects_per_sq_meter", 3.0)
def __init__(self, config): Module.__init__(self, config) self._replace_ratio = self.config.get_float("replace_ratio", 1) self._copy_properties = self.config.get_float("copy_properties", 1) self._max_tries = self.config.get_int("max_tries", 100000) self._objects_to_be_replaced = [] self._objects_to_replace_with = [] self._ignore_collision_with = []
def __init__(self, config): Module.__init__(self, config) self.postprocessing_modules_per_output = {} module_configs = config.get_raw_dict("postprocessing_modules", {}) for output_key in module_configs: self.postprocessing_modules_per_output[output_key] = Utility.initialize_modules(module_configs[output_key]) self.name_to_id = {} self.destination_frame = self.config.get_list("destination_frame", ["X", "Y", "Z"])
def __init__(self, config): Module.__init__(self, config) self._data_path = Utility.resolve_path( self.config.get_string("data_path")) self._img_files = glob.glob(os.path.join(self._data_path, "*.jpg")) self._num_used = self.config.get_int("num_used", 10) self._add_cp = self.config.get_raw_dict("add_custom_properties", {})
def __init__(self, config): Module.__init__(self, config) self.object_stopped_location_threshold = self.config.get_float( "object_stopped_location_threshold", 0.01) self.object_stopped_rotation_threshold = self.config.get_float( "object_stopped_rotation_threshold", 0.1) self.collision_margin = self.config.get_float("collision_margin", 0.001)
def __init__(self, config): Module.__init__(self, config) object_pose_sampler_config = config.get_raw_dict("object_pose_sampler", {}) camera_pose_sampler_config = config.get_raw_dict("camera_pose_sampler", {}) self._object_pose_sampler = Utility.initialize_modules([object_pose_sampler_config], {})[0] self._camera_pose_sampler = Utility.initialize_modules([camera_pose_sampler_config], {})[0]
def __init__(self, config): Module.__init__(self, config) self.postprocessing_modules_per_output = {} module_configs = config.get_raw_dict("postprocessing_modules", {}) for output_key in module_configs: self.postprocessing_modules_per_output[ output_key] = Utility.initialize_modules( module_configs[output_key], {})
def __init__(self, config): Module.__init__(self, config) self._folder_path = "" self._used_assets = [] self._add_cp = {} self._preload = False self._fill_used_empty_materials = False # makes the integration of complex materials easier addon_utils.enable("node_wrangler")
def __init__(self, config): Module.__init__(self, config) self._avoid_rendering = config.get_bool("avoid_rendering", False) self.postprocessing_modules_per_output = {} module_configs = config.get_raw_dict("postprocessing_modules", {}) for output_key in module_configs: self.postprocessing_modules_per_output[ output_key] = Utility.initialize_modules( module_configs[output_key])
def __init__(self, config): Module.__init__(self, config) # setting up the GlobalStorage global_config = Config(self.config.get_raw_dict("global", {})) GlobalStorage.init_global(global_config) # call the init again to make sure all values from the global config where read correctly, too self._default_init()
def __init__(self, config): Module.__init__(self, config) self.object_stopped_location_threshold = self.config.get_float("object_stopped_location_threshold", 0.01) self.object_stopped_rotation_threshold = self.config.get_float("object_stopped_rotation_threshold", 0.1) self.collision_margin = self.config.get_float("collision_margin", 0.001) self.collision_mesh_source = self.config.get_string('collision_mesh_source', 'FINAL') self.steps_per_sec = self.config.get_int("steps_per_sec", 60) self.solver_iters = self.config.get_int("solver_iters", 10) self.mass_scaling = self.config.get_bool("mass_scaling", False) self.mass_factor = self.config.get_float("mass_factor", 1)
def __init__(self, config): Module.__init__(self, config, False) camera_runner_config = config.get_raw_dict("camera_runner", {}) object_runners_config = config.get_list("object_runners", []) light_runners_config = config.get_list("light_runners", []) self._camera_runner = Utility.initialize_modules([camera_runner_config], {})[0] self._object_runners = Utility.initialize_modules(object_runners_config, {}) self._light_runners = Utility.initialize_modules(light_runners_config, {})
def __init__(self, config): Module.__init__(self, config) self.up_direction = config.get_vector3d( "up_direction", mathutils.Vector([0., 0., 1.])).normalized() self.min_distance = config.get_float("min_distance", 0.25) self.max_distance = config.get_float("max_distance", 0.6) self.placed_objects = [] self.surface = None self.surface_height = None
def __init__(self, config): Module.__init__(self, config) # Clean up example scene or scene created by last run when debugging pipeline inside blender Initializer.cleanup() # setting up the GlobalStorage global_config = Config(self.config.get_raw_dict("global", {})) GlobalStorage.init_global(global_config) # call the init again to make sure all values from the global config where read correctly, too self._default_init()
def __init__(self, config): Module.__init__(self, config) self.collision_margin = self.config.get_float("collision_margin", 0.001) self.collision_mesh_source = self.config.get_string('collision_mesh_source', 'FINAL') if config.has_param("steps_per_sec"): raise Exception("You are using the outdated parameter steps_per_sec. Please update your config by switching to substeps_per_frame (was changed in blender 2.91).") self.mass_scaling = self.config.get_bool("mass_scaling", False) self.mass_factor = self.config.get_float("mass_factor", 1) self.collision_shape = self.config.get_string("collision_shape", "CONVEX_HULL") self.friction = self.config.get_float("friction", 0.5) self.angular_damping = self.config.get_float("angular_damping", 0.1) self.linear_damping = self.config.get_float("linear_damping", 0.04) self.convex_decomposition_cache_path = self.config.get_string("convex_decomposition_cache_path", "resources/decomposition_cache")
def __init__(self, config): Module.__init__(self, config) self._avoid_rendering = config.get_bool("avoid_rendering", False) self.rgb_output_key = self.config.get_string("rgb_output_key", "colors") self.segmap_output_key = self.config.get_string( "segmap_output_key", "segmap") self.segcolormap_output_key = self.config.get_string( "segcolormap_output_key", "segcolormap") self._coco_data_dir = os.path.join(self._determine_output_dir(False), 'coco_data') if not os.path.exists(self._coco_data_dir): os.makedirs(self._coco_data_dir)
def __init__(self, config): Module.__init__(self, config) self.object_stopped_location_threshold = self.config.get_float("object_stopped_location_threshold", 0.01) self.object_stopped_rotation_threshold = self.config.get_float("object_stopped_rotation_threshold", 0.1) self.collision_margin = self.config.get_float("collision_margin", 0.001) self.collision_mesh_source = self.config.get_string('collision_mesh_source', 'FINAL') if config.has_param("steps_per_sec"): raise Exception("You are using the outdated parameter steps_per_sec. Please update your config by switching to substeps_per_frame (was changed in blender 2.91).") self.substeps_per_frame = self.config.get_int("substeps_per_frame", 10) self.solver_iters = self.config.get_int("solver_iters", 10) self.mass_scaling = self.config.get_bool("mass_scaling", False) self.mass_factor = self.config.get_float("mass_factor", 1) self.collision_shape = self.config.get_string("collision_shape", "CONVEX_HULL") self.friction = self.config.get_float("friction", 0.5) self.angular_damping = self.config.get_float("angular_damping",0.1) self.linear_damping = self.config.get_float("linear_damping",0.04)
def __init__(self, config): Module.__init__(self, config) self.randomization_level = self.config.get_float( "randomization_level", 0.2) self.randomize_textures_only = self.config.get_bool( "randomize_textures_only", False) self.output_textures_only = self.config.get_bool( "output_textures_only", True) self._objects_to_manipulate = None if self.config.has_param('manipulated_objects'): self._objects_to_manipulate = self.config.get_list( 'manipulated_objects') self._objects_to_extract_materials_from = None if self.config.has_param('objects_to_extract_mat'): self._objects_to_extract_materials_from = self.config.get_list( 'objects_to_extract_mat') self.scene_materials = []
def __init__(self, config): Module.__init__(self, config) self.object_stopped_location_threshold = self.config.get_float( "object_stopped_location_threshold", 0.01) self.object_stopped_rotation_threshold = self.config.get_float( "object_stopped_rotation_threshold", 0.1) self.collision_margin = self.config.get_float("collision_margin", 0.001) self.collision_mesh_source = self.config.get_string( 'collision_mesh_source', 'FINAL') self.steps_per_sec = self.config.get_int("steps_per_sec", 60) self.solver_iters = self.config.get_int("solver_iters", 10) self.mass_scaling = self.config.get_bool("mass_scaling", False) self.mass_factor = self.config.get_float("mass_factor", 1) self.collision_shape = self.config.get_string("collision_shape", "CONVEX_HULL") self.friction = self.config.get_float("friction", 0.5) self.angular_damping = self.config.get_float("angular_damping", 0.1) self.linear_damping = self.config.get_float("linear_damping", 0.04)
def __init__(self, config): Module.__init__(self, config) # Read in lights self.lights = {} # File format: <obj id> <number of lightbulb materials> <lightbulb material names> <number of lampshade materials> <lampshade material names> with open( Utility.resolve_path( os.path.join('resources', "suncg", "light_geometry_compact.txt"))) as f: lines = f.readlines() for row in lines: row = row.strip().split() self.lights[row[0]] = [[], []] index = 1 # Read in lightbulb materials number = int(row[index]) index += 1 for i in range(number): self.lights[row[0]][0].append(row[index]) index += 1 # Read in lampshade materials number = int(row[index]) index += 1 for i in range(number): self.lights[row[0]][1].append(row[index]) index += 1 # Read in windows self.windows = [] with open( Utility.resolve_path( os.path.join('resources', 'suncg', 'ModelCategoryMapping.csv')), 'r') as csvfile: reader = csv.DictReader(csvfile) for row in reader: if row["coarse_grained_class"] == "window": self.windows.append(row["model_id"]) self._collection_of_mats = {"lamp": {}, "window": {}, "ceiling": {}}
def __init__(self, config): Module.__init__(self, config)
def __init__(self, config): Module.__init__(self, config) self._avoid_rendering = config.get_bool("avoid_rendering", False) addon_utils.enable("render_auto_tile_size")
def __init__(self, config): Module.__init__(self, config) self.cross_source_settings = self.config.get_raw_dict( "cross_source_settings", {}) self.light_source_collection = ItemCollection( self._add_light_source, self.cross_source_settings)
def __init__(self, config): Module.__init__(self, config) addon_utils.enable("render_auto_tile_size")
def __init__(self, config): Module.__init__(self, config) self.source_frame = self.config.get_list("source_frame", ["X", "Y", "Z"])
def __init__(self, config): Module.__init__(self, config) for sys_path in self.config.get_list("sys_paths"): if 'bop_toolkit' in sys_path: sys.path.append(sys_path)