tracker-threshold

.vscode/c_cpp_properties.json

@@ -8,7 +8,7 @@
                 "/usr/local/cuda/include",
                 "/usr/local/include",
                 "/usr/include/**",
-                "/home/wangxin/projects/TensorRT-7.0.0/TensorRT/include",
+                "/home/wangxin/projects/TensorRT-7.0.0/TensorRT/include"
             ],
             "defines": [],
             "compilerPath": "/usr/bin/gcc",

CMakeLists.txt

@@ -16,8 +16,6 @@ include_directories(${OpenCV_INCLUDE_DIRS})
 link_directories(/usr/local/cuda/lib64)
 include_directories(/usr/local/cuda/include)
 
-link_directories(/usr/lib/x86_64-linux-gnu/)
-include_directories(/usr/include/x86_64-linux-gnu/)
 
 # eigen3
 find_package(Eigen3 REQUIRED)
@@ -37,9 +35,8 @@ link_directories(/root/TensorRT/TensorRT-7.0.0.11-cuda-10.0/lib)
     # include_directories(/usr/include/aarch64-linux-gnu)
 
 # yaml
-include_directories(/usr/local/include/yaml-cpp)
-# aarch64 -- figure out
-# include_directories(/usr/include/yaml-cpp)
+# include_directories(/usr/local/include/yaml-cpp)
+include_directories(/usr/include/yaml-cpp)
 
 # glog gflags
 # sudo apt-get install libgoogle-glog*

configs/tasks/thermal/yolov5.yaml

@@ -77,6 +77,7 @@ units:
         match_thresh: 0.8
         # output detection after it was tracked in x frames
         init_frames: 0
+        out_lost_threshold: 3
     -
         name: ByteTracker
         inputNames: [detections2]

src/libs/trackers/byte_tracker.cpp

@@ -19,7 +19,7 @@ bool ByteTracker::Init(YAML::Node& node)
 	high_thresh = node["high_thresh"].as<float>();
 	match_thresh = node["match_thresh"].as<float>();
 	init_frames = node["init_frames"].as<int>();
-	if(!node["out_lost_threshold"].IsNull()){
+	if(node["out_lost_threshold"].IsDefined() && !node["out_lost_threshold"].IsNull()){
 		out_lost_threshold = node["out_lost_threshold"].as<int>();
 	}