From 51fa651bc73387aab84679856b9f58aa8b5a3740 Mon Sep 17 00:00:00 2001
From: Jonathan Rynne <jrynne@arcturusnetworks.com>
Date: Thu, 6 Mar 2025 12:24:42 -0500
Subject: [PATCH] Add missing namespaces

---
 deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h | 11 +++++++++--
 deploy/TensorRT/cpp/include/bytetrack/STrack.h      |  9 ++++++---
 deploy/TensorRT/cpp/src/STrack.cpp                  |  6 +++++-
 3 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h b/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h
index c9bb696a..06aaf416 100644
--- a/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h
+++ b/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h
@@ -2,6 +2,15 @@
 
 #include "STrack.h"
 
+// struct Object
+// {
+//     cv::Rect_<float> rect;
+//     int label;
+//     float prob;
+// };
+
+namespace bytetrack {
+
 struct Object
 {
     cv::Rect_<float> rect;
@@ -9,8 +18,6 @@ struct Object
     float prob;
 };
 
-namespace bytetrack {
-
 class BYTETracker
 {
 public:
diff --git a/deploy/TensorRT/cpp/include/bytetrack/STrack.h b/deploy/TensorRT/cpp/include/bytetrack/STrack.h
index 4bd9d000..3c76cd16 100644
--- a/deploy/TensorRT/cpp/include/bytetrack/STrack.h
+++ b/deploy/TensorRT/cpp/include/bytetrack/STrack.h
@@ -6,10 +6,11 @@
 using namespace cv;
 using namespace std;
 
+namespace bytetrack {
+
 enum TrackState { New = 0, Tracked, Lost, Removed };
 
-class STrack
-{
+class STrack {
 public:
 	STrack(vector<float> tlwh_, float score);
 	~STrack();
@@ -47,4 +48,6 @@ class STrack
 
 private:
 	byte_kalman::KalmanFilter kalman_filter;
-};
\ No newline at end of file
+};
+
+}  // namespace bytetrack
\ No newline at end of file
diff --git a/deploy/TensorRT/cpp/src/STrack.cpp b/deploy/TensorRT/cpp/src/STrack.cpp
index ceb08c7c..78b6575d 100644
--- a/deploy/TensorRT/cpp/src/STrack.cpp
+++ b/deploy/TensorRT/cpp/src/STrack.cpp
@@ -1,5 +1,7 @@
 #include "STrack.h"
 
+namespace bytetrack {
+
 STrack::STrack(vector<float> tlwh_, float score)
 {
 	_tlwh.resize(4);
@@ -189,4 +191,6 @@ void STrack::multi_predict(vector<STrack*> &stracks, byte_kalman::KalmanFilter &
 		}
 		kalman_filter.predict(stracks[i]->mean, stracks[i]->covariance);
 	}
-}
\ No newline at end of file
+}
+
+}  // namespace bytetrack
\ No newline at end of file