diff --git a/butia_image2world/scripts/butia_image2world/image2world/image2world.py b/butia_image2world/scripts/butia_image2world/image2world/image2world.py index 36700dc..6ef94e1 100755 --- 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/butia_recognition/scripts/butia_recognition/yolo_tracker_recognition/yolo_tracker_recognition.py b/butia_recognition/scripts/butia_recognition/yolo_tracker_recognition/yolo_tracker_recognition.py index 66e5e30..b13a261 100755 --- a/butia_recognition/scripts/butia_recognition/yolo_tracker_recognition/yolo_tracker_recognition.py +++ b/butia_recognition/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 index 8261f52..99bda27 160000 --- a/butia_vision_msgs +++ b/butia_vision_msgs @@ -1 +1 @@ -Subproject commit 8261f52546e06231c2fd1987a55046e1e75ee68c +Subproject commit 99bda273754c2b7cd8e4a9cad06ce3f0cb8ee9bb