floraachy
/
SpireCV
forked from floratest1/SpireCV
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

This commit is contained in:
jario 2024-01-19 22:55:31 +08:00
parent 212c9f18fa
commit 9c4d8a400c
1 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
algorithm/common_det/cuda/common_det_cuda_impl.cpp
View File

@ -1,6 +1,7 @@
#include "common_det_cuda_impl.h" #include "common_det_cuda_impl.h"
#include <cmath> #include <cmath>
#include <fstream> #include <fstream>
#include <iostream>
#include "sv_util.h" #include "sv_util.h"
#define SV_MODEL_DIR "/SpireCV/models/" #define SV_MODEL_DIR "/SpireCV/models/"
@ -366,7 +367,7 @@ bool CommonObjectDetectorCUDAImpl::cudaSetup(CommonObjectDetectorBase* base_, bo
std::string model = base_->getModel(); std::string model = base_->getModel();
int bs = base_->getBatchSize(); int bs = base_->getBatchSize();
char bs_c[8]; char bs_c[8];
itoa(bs, bs_c, 10); sprintf(bs_c, "%d", bs);
std::string bs_s(bs_c); std::string bs_s(bs_c);
std::string engine_fn = get_home() + SV_MODEL_DIR + dataset + ".engine"; std::string engine_fn = get_home() + SV_MODEL_DIR + dataset + ".engine";
@ -449,17 +450,5 @@ bool CommonObjectDetectorCUDAImpl::cudaSetup(CommonObjectDetectorBase* base_, bo
return false; return false;
} }
} }