diff --git a/code/perception/src/vision_node.py b/code/perception/src/vision_node.py index 804932b6..0f4f0448 100755 --- a/code/perception/src/vision_node.py +++ b/code/perception/src/vision_node.py @@ -395,10 +395,10 @@ def predict_ultralytics(self, image): image_np_with_detections = torch.tensor(transposed_image, dtype=torch.uint8) - if 9 in output[0].boxes.cls: - asyncio.run(self.process_traffic_lights(output[0], - cv_image, - image.header)) + if 9 in output[0].boxes.cls: + asyncio.run(self.process_traffic_lights(output[0], + cv_image, + image.header)) c_boxes = torch.stack(c_boxes) # print(image_np_with_detections.shape, c_boxes.shape, c_labels)