added bbox2D to description3D

butia-bots · Jul 3, 2023 · 21de53d · 21de53d
1 parent 31d1795
commit 21de53d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/butia_image2world/scripts/butia_image2world/image2world/image2world.py b/butia_image2world/scripts/butia_image2world/image2world/image2world.py
@@ -320,9 +320,11 @@ def __poseDescriptionProcessing(self, data, description2d : Description2D, heade
         return description3d
 
 
-    def __createDescription3D(self, source_data, description2d, header):
+    def __createDescription3D(self, source_data, description2d : Description2D, header):
         if description2d.type in self.DESCRIPTION_PROCESSING_ALGORITHMS:
-            return self.DESCRIPTION_PROCESSING_ALGORITHMS[description2d.type](source_data, description2d, header)
+            description3D : Description3D = self.DESCRIPTION_PROCESSING_ALGORITHMS[description2d.type](source_data, description2d, header)
+            description3D.bbox2D = description2d.bbox
+            return description3D
         else:
             return None