diff --git a/src/Detector/tensorrt_onnx/YoloONNXv26_bb.hpp b/src/Detector/tensorrt_onnx/YoloONNXv26_bb.hpp index 7677244b..19cdd67a 100644 --- a/src/Detector/tensorrt_onnx/YoloONNXv26_bb.hpp +++ b/src/Detector/tensorrt_onnx/YoloONNXv26_bb.hpp @@ -43,7 +43,7 @@ class YOLOv26_bb_onnx : public YoloONNX auto ind = i * m_outpuDims[0].d[2]; float classConf = output[ind + 4]; - int64_t classId = output[ind + 5]; + int classId = static_cast(output[ind + 5]); if (classConf >= m_params.m_confThreshold) { diff --git a/src/Detector/tensorrt_onnx/YoloONNXv26_instance.hpp b/src/Detector/tensorrt_onnx/YoloONNXv26_instance.hpp index abcb3b4f..9ec2d27e 100644 --- a/src/Detector/tensorrt_onnx/YoloONNXv26_instance.hpp +++ b/src/Detector/tensorrt_onnx/YoloONNXv26_instance.hpp @@ -65,7 +65,7 @@ class YOLOv26_instance_onnx : public YoloONNX size_t k = i * dimensions; float objectConf = output[k + 4]; - int classId = output[k + 5]; + int classId = static_cast(output[k + 5]); if (objectConf >= m_params.m_confThreshold) { diff --git a/src/Detector/tensorrt_onnx/YoloONNXv26_obb.hpp b/src/Detector/tensorrt_onnx/YoloONNXv26_obb.hpp index 0b70b509..8a097f2f 100644 --- a/src/Detector/tensorrt_onnx/YoloONNXv26_obb.hpp +++ b/src/Detector/tensorrt_onnx/YoloONNXv26_obb.hpp @@ -43,7 +43,7 @@ class YOLOv26_obb_onnx : public YoloONNX auto ind = i * m_outpuDims[0].d[2]; float classConf = output[ind + 4]; - int64_t classId = output[ind + 5]; + int classId = static_cast(output[ind + 5]); if (classConf >= m_params.m_confThreshold) { diff --git a/src/Tracker/TrackerSettings.cpp b/src/Tracker/TrackerSettings.cpp index e76dd72e..e06f2311 100644 --- a/src/Tracker/TrackerSettings.cpp +++ b/src/Tracker/TrackerSettings.cpp @@ -57,12 +57,12 @@ bool ParseTrackerSettings(const std::string& settingsFile, TrackerSettings& trac trackerSettings.m_useAbandonedDetection = reader.GetInteger("tracking", "detect_abandoned", 0) != 0; trackerSettings.m_minStaticTime = reader.GetInteger("tracking", "min_static_time", 5); trackerSettings.m_maxStaticTime = reader.GetInteger("tracking", "max_static_time", 25); - trackerSettings.m_maxSpeedForStatic = reader.GetInteger("tracking", "max_speed_for_static", 10); + trackerSettings.m_maxSpeedForStatic = static_cast(reader.GetReal("tracking", "max_speed_for_static", 0.5)); trackerSettings.m_byteTrackSettings.m_trackBuffer = reader.GetInteger("tracking", "bytetrack_track_buffer", 30); - trackerSettings.m_byteTrackSettings.m_trackThresh = reader.GetReal("tracking", "bytetrack_track_thresh", 0.5); - trackerSettings.m_byteTrackSettings.m_highThresh = reader.GetReal("tracking", "bytetrack_high_thresh", 0.5); - trackerSettings.m_byteTrackSettings.m_matchThresh = reader.GetReal("tracking", "bytetrack_match_thresh", 0.8); + trackerSettings.m_byteTrackSettings.m_trackThresh = static_cast(reader.GetReal("tracking", "bytetrack_track_thresh", 0.5)); + trackerSettings.m_byteTrackSettings.m_highThresh = static_cast(reader.GetReal("tracking", "bytetrack_high_thresh", 0.5)); + trackerSettings.m_byteTrackSettings.m_matchThresh = static_cast(reader.GetReal("tracking", "bytetrack_match_thresh", 0.8)); // Read detection settings trackerSettings.m_nnWeights = reader.GetString("detection", "nn_weights", "data/yolov4-tiny_best.weights"); diff --git a/src/Tracker/byte_track/BYTETracker.cpp b/src/Tracker/byte_track/BYTETracker.cpp index 8a63cf56..144490b1 100644 --- a/src/Tracker/byte_track/BYTETracker.cpp +++ b/src/Tracker/byte_track/BYTETracker.cpp @@ -310,8 +310,8 @@ void byte_track::BYTETracker::removeDuplicateStracks(const std::vector a_overlapping(a_stracks.size(), false), b_overlapping(b_stracks.size(), false); for (const auto &[a_idx, b_idx] : overlapping_combinations) { - const int timep = a_stracks[a_idx]->getFrameId() - a_stracks[a_idx]->getStartFrameId(); - const int timeq = b_stracks[b_idx]->getFrameId() - b_stracks[b_idx]->getStartFrameId(); + const size_t timep = a_stracks[a_idx]->getFrameId() - a_stracks[a_idx]->getStartFrameId(); + const size_t timeq = b_stracks[b_idx]->getFrameId() - b_stracks[b_idx]->getStartFrameId(); if (timep > timeq) b_overlapping[b_idx] = true; else @@ -359,16 +359,9 @@ void byte_track::BYTETracker::linearAssignment(const std::vector= 0) - { - std::vector match; - match.push_back(i); - match.push_back(rowsol[i]); - matches.push_back(match); - } + matches.push_back({ (int)i, rowsol[i] }); else - { a_unmatched.push_back(i); - } } for (size_t i = 0; i < colsol.size(); i++)