ultralytics upgraded

mgonzs13 · Jun 28, 2024 · e6ae844 · e6ae844
1 parent 7763938
commit e6ae844
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
 opencv-python==4.8.1.78
 typing-extensions>=4.4.0
-ultralytics==8.2.38
+ultralytics==8.2.45
 lap==0.4.0
diff --git a/yolov8_ros/yolov8_ros/yolov8_node.py b/yolov8_ros/yolov8_ros/yolov8_node.py
@@ -27,7 +27,7 @@
 
 from cv_bridge import CvBridge
 
-from torch import cuda
+import torch
 from ultralytics import YOLO
 from ultralytics.engine.results import Results
 from ultralytics.engine.results import Boxes
@@ -125,7 +125,7 @@ def on_deactivate(self, state: LifecycleState) -> TransitionCallbackReturn:
         del self.yolo
         if "cuda" in self.device:
             self.get_logger().info("Clearing CUDA cache")
-            cuda.empty_cache()
+            torch.cuda.empty_cache()
 
         self.destroy_subscription(self._sub)
         self._sub = None