fix cpp infer (#1682)

This commit is contained in:
littletomatodonkey
2021-01-07 19:27:22 +08:00
committed by GitHub
parent a27bd4d5d1
commit 9336e34430
5 changed files with 8 additions and 9 deletions
+1 -1
View File
@@ -40,7 +40,7 @@ public:
const int &gpu_id, const int &gpu_mem,
const int &cpu_math_library_num_threads,
const bool &use_mkldnn, const double &cls_thresh,
const bool &use_tensorrt, const &bool use_fp16) {
const bool &use_tensorrt, const bool &use_fp16) {
this->use_gpu_ = use_gpu;
this->gpu_id_ = gpu_id;
this->gpu_mem_ = gpu_mem;
+2 -3
View File
@@ -44,9 +44,8 @@ public:
const bool &use_mkldnn, const int &max_side_len,
const double &det_db_thresh,
const double &det_db_box_thresh,
const double &det_db_unclip_ratio,
const bool &visualize const bool &use_tensorrt,
const bool &use_fp16) {
const double &det_db_unclip_ratio, const bool &visualize,
const bool &use_tensorrt, const bool &use_fp16) {
this->use_gpu_ = use_gpu;
this->gpu_id_ = gpu_id;
this->gpu_mem_ = gpu_mem;