diff --git a/blenderproc/python/modules/writer/BopWriter.py b/blenderproc/python/modules/writer/BopWriter.py index 74cb986d6..8fde2591f 100644 --- a/blenderproc/python/modules/writer/BopWriter.py +++ b/blenderproc/python/modules/writer/BopWriter.py @@ -1,6 +1,5 @@ from blenderproc.python.modules.writer.WriterInterface import WriterInterface from blenderproc.python.writer.BopWriterUtility import write_bop -from blenderproc.python.type.MeshObjectUtility import MeshObject import os @@ -69,7 +68,6 @@ def run(self): if self._avoid_output: print("Avoid output is on, no output produced!") else: - MeshObject.get_all_mesh_objects() write_bop(output_dir = os.path.join(self._determine_output_dir(False), 'bop_data'), dataset = self._dataset, append_to_existing_output = self._append_to_existing_output, diff --git a/examples/basics/camera_object_pose/README.md b/examples/basics/camera_object_pose/README.md index 88b338c18..1dad01cb8 100644 --- a/examples/basics/camera_object_pose/README.md +++ b/examples/basics/camera_object_pose/README.md @@ -80,7 +80,7 @@ bproc.camera.add_camera_pose(cam2world) data = bproc.renderer.render() # Write object poses, color and depth in bop format -bproc.writer.write_bop(args.output_dir, data["depth"], data["colors"], m2mm=True, append_to_existing_output=True) +bproc.writer.write_bop(args.output_dir, [obj], data["depth"], data["colors"], m2mm=True, append_to_existing_output=True) ``` * Saves all pose and camera information that is provided in BOP datasets. diff --git a/examples/basics/camera_object_pose/main.py b/examples/basics/camera_object_pose/main.py index 3faf6d65b..f99ba1496 100644 --- a/examples/basics/camera_object_pose/main.py +++ b/examples/basics/camera_object_pose/main.py @@ -60,4 +60,4 @@ data = bproc.renderer.render() # Write object poses, color and depth in bop format -bproc.writer.write_bop(args.output_dir, data["depth"], data["colors"], m2mm=True, append_to_existing_output=True) +bproc.writer.write_bop(args.output_dir, [obj], data["depth"], data["colors"], m2mm=True, append_to_existing_output=True)