Merge pull request #1086 from matteomastrogiuseppe/main

cornerfarmer · web-flow · commit ec5493179740 · 2024-04-10T11:22:23.000+02:00
fix pools termination in BOP dataset generation
diff --git a/blenderproc/python/writer/BopWriterUtility.py b/blenderproc/python/writer/BopWriterUtility.py
@@ -179,6 +179,10 @@ def write_bop(output_dir: str, target_objects: Optional[List[MeshObject]] = None
 
         _BopWriterUtility.calc_gt_coco(chunk_dirs=chunk_dirs, dataset_objects=dataset_objects,
                                        starting_frame_id=starting_frame_id)
+        
+        pool.close()
+        pool.join()
+
 
 
 def bop_pose_to_pyrender_coordinate_system(cam_R_m2c: np.ndarray, cam_t_m2c: np.ndarray) -> np.ndarray:
