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