Fix(camera_object_pose): fct arguments

ZuzkaU · Nov 27, 2021 · 46db849 · 46db849
1 parent 7907269
commit 46db849
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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)