From 9c4d8a400c2b9f43b1e8821464b9696872c84b8b Mon Sep 17 00:00:00 2001 From: jario Date: Fri, 19 Jan 2024 22:55:31 +0800 Subject: [PATCH] fix --- .../common_det/cuda/common_det_cuda_impl.cpp | 15 ++------------- 1 file changed, 2 insertions(+), 13 deletions(-) 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; } - - - - - - - - - - - - }