Merge pull request ifzhang#158 from XiaoPeilun/main

fix c++ bug: update tlbr after kalman prediction
qfgaohao · Apr 27, 2022 · f137cbf · f137cbf
2 parents 7a5507c + b500c6f
commit f137cbf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/deploy/TensorRT/cpp/src/STrack.cpp b/deploy/TensorRT/cpp/src/STrack.cpp
@@ -188,5 +188,7 @@ void STrack::multi_predict(vector<STrack*> &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();
 	}
 }
diff --git a/deploy/ncnn/cpp/src/STrack.cpp b/deploy/ncnn/cpp/src/STrack.cpp
@@ -188,5 +188,7 @@ void STrack::multi_predict(vector<STrack*> &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();
 	}
 }