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 run(self): horizon_color = self.config.get_list("horizon_color", [0.05, 0.05, 0.05]) compute_device = self.config.get_string("compute_device", "GPU") compute_device_type = self.config.get_string("compute_device_type", None) use_experimental_features = self.config.get_bool( "use_experimental_features", False) Initializer.init(horizon_color, compute_device, compute_device_type, use_experimental_features, clean_up_scene=False)
from src.utility.RendererUtility import RendererUtility import numpy as np import argparse import os parser = argparse.ArgumentParser() parser.add_argument( 'house', help="Path to the house.json file of the SUNCG scene to load") parser.add_argument('output_dir', nargs='?', default="examples/datasets/suncg_with_cam_sampling/output", help="Path to where the final files, will be saved") args = parser.parse_args() Initializer.init() # load the objects into the scene label_mapping = LabelIdMapping.from_csv( Utility.resolve_path( os.path.join('resources', 'id_mappings', 'nyu_idset.csv'))) objs = SuncgLoader.load(args.house, label_mapping) # makes Suncg objects emit light SuncgLighting.light() # Init sampler for sampling locations inside the loaded suncg house point_sampler = SuncgPointInRoomSampler(objs) # Init bvh tree containing all mesh objects bvh_tree = MeshObject.create_bvh_tree_multi_objects( [o for o in objs if isinstance(o, MeshObject)])