dealing with class number

butia-bots · Jul 6, 2023 · e499441 · e499441
1 parent 23b1e72
commit e499441
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/butia_image2world/scripts/butia_image2world/image2world/image2world.py b/butia_image2world/scripts/butia_image2world/image2world/image2world.py
@@ -324,6 +324,8 @@ def __createDescription3D(self, source_data, description2d : Description2D, head
         if description2d.type in self.DESCRIPTION_PROCESSING_ALGORITHMS:
             description3D : Description3D = self.DESCRIPTION_PROCESSING_ALGORITHMS[description2d.type](source_data, description2d, header)
             description3D.bbox2D = description2d.bbox
+            description3D.class_num = description2d.class_num
+            print(description3D.class_num)
             return description3D
         else:
             return None

diff --git a/...ecognition/scripts/butia_recognition/yolo_tracker_recognition/yolo_tracker_recognition.py b/...ecognition/scripts/butia_recognition/yolo_tracker_recognition/yolo_tracker_recognition.py
@@ -156,6 +156,7 @@ def callback(self, *args):
             description.label = self.model.names[clss]
             description.type = Description2D.DETECTION
             description.score = score
+            description.class_num = clss
             description.id = i
 
             box_label = ""

diff --git a/butia_vision_msgs b/butia_vision_msgs