diff --git a/algorithm/common_det/cuda/common_det_cuda_impl.cpp b/algorithm/common_det/cuda/common_det_cuda_impl.cpp index 11b24a4..40056ea 100644 --- a/algorithm/common_det/cuda/common_det_cuda_impl.cpp +++ b/algorithm/common_det/cuda/common_det_cuda_impl.cpp @@ -1,6 +1,7 @@ #include "common_det_cuda_impl.h" #include #include +#include #include "sv_util.h" #define SV_MODEL_DIR "/SpireCV/models/" @@ -366,7 +367,7 @@ bool CommonObjectDetectorCUDAImpl::cudaSetup(CommonObjectDetectorBase* base_, bo std::string model = base_->getModel(); int bs = base_->getBatchSize(); char bs_c[8]; - itoa(bs, bs_c, 10); + sprintf(bs_c, "%d", bs); std::string bs_s(bs_c); std::string engine_fn = get_home() + SV_MODEL_DIR + dataset + ".engine"; @@ -449,17 +450,5 @@ bool CommonObjectDetectorCUDAImpl::cudaSetup(CommonObjectDetectorBase* base_, bo return false; } - - - - - - - - - - - - }