From 3e569ce80590f782e594bd8fbbf921e0a412eadd Mon Sep 17 00:00:00 2001 From: MaxJa4 <74194322+MaxJa4@users.noreply.github.com> Date: Sun, 10 Mar 2024 21:36:09 +0100 Subject: [PATCH] Fix tab --- code/perception/src/vision_node.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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)