Fix tab

una-auxme · Mar 10, 2024 · 3e569ce · 3e569ce
1 parent fd6ac66
commit 3e569ce
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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)