mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Fix a bug with batch > 1

Browse Source 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2021-04-30 22:26:33 +02:00
parent 2367519799
commit ff6e0e010a
2 changed files with 99 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/CenternetDetection3DTrack.h
+4 -5
View File
@@ -148,10 +148,9 @@ private:
std::vector<struct detectionRes> det_res;
int count_det;
//tracks
std::vector<struct trackingRes> tr_res;
std::vector<std::vector<struct trackingRes>> batchTracked;
int count_tr;
int track_id=0;
std::vector<std::vector<struct trackingRes>> tr_res;
std::vector<int> count_tr;
std::vector<int> track_id;
bool init_preprocessing();
@@ -161,7 +160,7 @@ private:
void pre_inf(const int bi);
void _get_additional_inputs();
cv::Mat transform_preds_with_trans(float x1, float x2);
void tracking();
void tracking(const int bi);
public:
tk::dnn::Network *pre_phase_net = nullptr;