From b500c6f3abd7d536b8786f9c7ead33b4c4c13855 Mon Sep 17 00:00:00 2001 From: XiaoPeilun Date: Sun, 6 Mar 2022 00:06:20 +0800 Subject: [PATCH] fix c++ bug: update tlbr after kalman prediction --- deploy/TensorRT/cpp/src/STrack.cpp | 2 ++ deploy/ncnn/cpp/src/STrack.cpp | 2 ++ 2 files changed, 4 insertions(+) diff --git a/deploy/TensorRT/cpp/src/STrack.cpp b/deploy/TensorRT/cpp/src/STrack.cpp index ceb08c7c..0e8653e8 100644 --- a/deploy/TensorRT/cpp/src/STrack.cpp +++ b/deploy/TensorRT/cpp/src/STrack.cpp @@ -188,5 +188,7 @@ void STrack::multi_predict(vector &stracks, byte_kalman::KalmanFilter & stracks[i]->mean[7] = 0; } kalman_filter.predict(stracks[i]->mean, stracks[i]->covariance); + stracks[i]->static_tlwh(); + stracks[i]->static_tlbr(); } } \ No newline at end of file diff --git a/deploy/ncnn/cpp/src/STrack.cpp b/deploy/ncnn/cpp/src/STrack.cpp index ceb08c7c..0e8653e8 100644 --- a/deploy/ncnn/cpp/src/STrack.cpp +++ b/deploy/ncnn/cpp/src/STrack.cpp @@ -188,5 +188,7 @@ void STrack::multi_predict(vector &stracks, byte_kalman::KalmanFilter & stracks[i]->mean[7] = 0; } kalman_filter.predict(stracks[i]->mean, stracks[i]->covariance); + stracks[i]->static_tlwh(); + stracks[i]->static_tlbr(); } } \ No newline at end of file