未验证 提交 9b6f1205 编写于 作者: F Feng Ni 提交者: GitHub

fix cpp infer of jdetracker (#6502)

上级 2b34bd76
...@@ -58,8 +58,8 @@ bool JDETracker::update(const cv::Mat &dets, const cv::Mat &emb, std::vector<Tra ...@@ -58,8 +58,8 @@ bool JDETracker::update(const cv::Mat &dets, const cv::Mat &emb, std::vector<Tra
TrajectoryPool candidates(dets.rows); TrajectoryPool candidates(dets.rows);
for (int i = 0; i < dets.rows; ++i) for (int i = 0; i < dets.rows; ++i)
{ {
float score = *dets.ptr<float>(i, 4); float score = *dets.ptr<float>(i, 1);
const cv::Mat &ltrb_ = dets(cv::Rect(0, i, 4, 1)); const cv::Mat &ltrb_ = dets(cv::Rect(2, i, 4, 1));
cv::Vec4f ltrb = mat2vec4f(ltrb_); cv::Vec4f ltrb = mat2vec4f(ltrb_);
const cv::Mat &embedding = emb(cv::Rect(0, i, emb.cols, 1)); const cv::Mat &embedding = emb(cv::Rect(0, i, emb.cols, 1));
candidates[i] = Trajectory(ltrb, score, embedding); candidates[i] = Trajectory(ltrb, score, embedding);
......
...@@ -56,8 +56,8 @@ bool JDETracker::update(const cv::Mat &dets, ...@@ -56,8 +56,8 @@ bool JDETracker::update(const cv::Mat &dets,
++timestamp; ++timestamp;
TrajectoryPool candidates(dets.rows); TrajectoryPool candidates(dets.rows);
for (int i = 0; i < dets.rows; ++i) { for (int i = 0; i < dets.rows; ++i) {
float score = *dets.ptr<float>(i, 4); float score = *dets.ptr<float>(i, 1);
const cv::Mat &ltrb_ = dets(cv::Rect(0, i, 4, 1)); const cv::Mat &ltrb_ = dets(cv::Rect(2, i, 4, 1));
cv::Vec4f ltrb = mat2vec4f(ltrb_); cv::Vec4f ltrb = mat2vec4f(ltrb_);
const cv::Mat &embedding = emb(cv::Rect(0, i, emb.cols, 1)); const cv::Mat &embedding = emb(cv::Rect(0, i, emb.cols, 1));
candidates[i] = Trajectory(ltrb, score, embedding); candidates[i] = Trajectory(ltrb, score, embedding);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册