Method load_obj

blenderproc/python/loader/SuncgLoader.py:281–314

Load the wavefront object file from the given path and adjust according to the given arguments. :param path: The path to the .obj file. :param metadata: A dict of metadata which will be written into the object's custom data. :param material_adjustments: Adjustments to the m

(path: str, metadata: Dict[str, Union[str, int]], material_adjustments: List[Dict[str, str]],
                 transform: Optional[Matrix] = None, parent: Optional[Entity] = None)

    @staticmethod
    def load_obj(path: str, metadata: Dict[str, Union[str, int]], material_adjustments: List[Dict[str, str]],
                 transform: Optional[Matrix] = None, parent: Optional[Entity] = None) -> List[MeshObject]:
        """ Load the wavefront object file from the given path and adjust according to the given arguments.

        :param path: The path to the .obj file.
        :param metadata: A dict of metadata which will be written into the object's custom data.
        :param material_adjustments: Adjustments to the materials which were specified inside house.json.
        :param transform: The transformation that should be applied to the loaded objects.
        :param parent: The parent object to which the object should be linked
        :return: The list of loaded mesh objects.
        """
        if not os.path.exists(path):
            print(f"Warning: {path} is missing!")
            return []
        object_already_loaded = path in _SuncgLoader.collection_of_loaded_objs
        loaded_objects = load_obj(filepath=path, cached_objects=_SuncgLoader.collection_of_loaded_objs)
        if object_already_loaded:
            print(f"Duplicate object: {path}")
            for obj in loaded_objects:
                # the original object matrix from the .obj loader -> is not an identity matrix
                obj.set_local2world_mat(Matrix([[1, 0, 0, 0], [0, 0, -1, 0], [0, 1, 0, 0], [0, 0, 0, 1]]))
                # remove all custom properties
                obj.clear_all_cps()
        # Go through all imported objects
        for obj in loaded_objects:
            for key in metadata.keys():
                used_key = key
                if key == "type":
                    used_key = "suncg_type"
                obj.set_cp(used_key, metadata[key])

            _SuncgLoader.transform_and_colorize_object(obj, material_adjustments, transform, parent)

        return loaded_objects

    @staticmethod
    def transform_and_colorize_object(obj: MeshObject, material_adjustments: List[Dict[str, str]],