From 305928b882a28574efe5ece8eeaf410f1e8f9e4d Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Sun, 7 Nov 2021 20:02:03 +0800 Subject: [PATCH] [MOT] fix stream predict (#4497) --- deploy/pptracking/include/pipeline.h | 2 +- deploy/pptracking/src/pipeline.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/deploy/pptracking/include/pipeline.h b/deploy/pptracking/include/pipeline.h index dd1d00491..66aa84ac7 100644 --- a/deploy/pptracking/include/pipeline.h +++ b/deploy/pptracking/include/pipeline.h @@ -79,7 +79,7 @@ class Pipeline { void PredictMTMCT(const std::vector video_inputs); // Run pipeline in stream - void RunMOTStream(const cv::Mat img, const int frame_id, cv::Mat out_img, std::vector& records, std::vector& count_list, std::vector& in_count_list, std::vector& out_count_list); + void RunMOTStream(const cv::Mat img, const int frame_id, cv::Mat& out_img, std::vector& records, std::vector& count_list, std::vector& in_count_list, std::vector& out_count_list); void RunMTMCTStream(const std::vector imgs, std::vector& records); void PrintBenchmarkLog(std::vector det_time, int img_num); diff --git a/deploy/pptracking/src/pipeline.cc b/deploy/pptracking/src/pipeline.cc index e36add5c9..189630a09 100644 --- a/deploy/pptracking/src/pipeline.cc +++ b/deploy/pptracking/src/pipeline.cc @@ -214,7 +214,7 @@ void Pipeline::PredictMTMCT(const std::vector video_path) { throw "Not Implement!"; } -void Pipeline::RunMOTStream(const cv::Mat img, const int frame_id, cv::Mat out_img, std::vector& records, std::vector& count_list, std::vector& in_count_list, std::vector& out_count_list) { +void Pipeline::RunMOTStream(const cv::Mat img, const int frame_id, cv::Mat& out_img, std::vector& records, std::vector& count_list, std::vector& in_count_list, std::vector& out_count_list) { PaddleDetection::MOTResult result; std::vector det_times(3); double times; -- GitLab