set default ultralytics device to cuda:0 (#111)

* change ultralytics device to 'cuda:0' * add model files to .gitignore
itu-auv · Dec 26, 2024 · 44893fe · 44893fe
1 parent bfc3560
commit 44893fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -37,3 +37,6 @@
 
 *__pycache__*
 *.pyc
+
+# ultralytics models
+*.pt
diff --git a/auv_vision/auv_detection/launch/tracker.launch b/auv_vision/auv_detection/launch/tracker.launch
@@ -12,7 +12,7 @@
   <arg name="max_det" default="300"/>
   <arg name="classes" default=""/>
   <arg name="tracker" default="bytetrack.yaml"/>
-  <arg name="device" default="cpu"/>
+  <arg name="device" default="cuda:0"/>
   <arg name="result_conf" default="true"/>
   <arg name="result_line_width" default="1"/>
   <arg name="result_font_size" default="1"/>
-Original file line number
+Diff line change
@@ Expand Up / @@ -37,3 +37,6 @@ @@
     *__pycache__*
     *.pyc
+    # ultralytics models
+    *.pt