From 44893fe6c77205c1874876a0821c7de4690b672b Mon Sep 17 00:00:00 2001
From: Talha <talha@karasu.me>
Date: Thu, 26 Dec 2024 14:04:19 +0200
Subject: [PATCH] set default ultralytics device to `cuda:0` (#111)

* change ultralytics device to 'cuda:0'

* add model files to .gitignore
---
 .gitignore                                     | 3 +++
 auv_vision/auv_detection/launch/tracker.launch | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index a53506f9..f6d79cf5 100644
--- 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
index a5ad4cdf..a3d3a183 100644
--- 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"/>