@@ -62,8 +62,9 @@ TrtRTMDet::TrtRTMDet(
62
62
const autoware::tensorrt_common::BuildConfig & build_config,
63
63
const std::string & calibration_image_list_file_path, const double norm_factor,
64
64
const std::vector<float > & mean, const std::vector<float > & std,
65
- [[maybe_unused]] const std::string & cache_dir, const autoware::tensorrt_common::BatchConfig & batch_config,
66
- const size_t max_workspace_size, const std::vector<std::string> & plugin_paths)
65
+ [[maybe_unused]] const std::string & cache_dir,
66
+ const autoware::tensorrt_common::BatchConfig & batch_config, const size_t max_workspace_size,
67
+ const std::vector<std::string> & plugin_paths)
67
68
: score_threshold_{score_threshold},
68
69
nms_threshold_{nms_threshold},
69
70
mask_threshold_{mask_threshold},
@@ -190,8 +191,8 @@ void TrtRTMDet::preprocess_gpu(const std::vector<cv::Mat> & images)
190
191
scale_height_ = input_height / static_cast <float >(image.rows );
191
192
image_buf_h_ = autoware::cuda_utils::make_unique_host<unsigned char []>(
192
193
image.cols * image.rows * 3 * batch_size, cudaHostAllocWriteCombined);
193
- image_buf_d_ =
194
- autoware::cuda_utils::make_unique< unsigned char []>( image.cols * image.rows * 3 * batch_size);
194
+ image_buf_d_ = autoware::cuda_utils::make_unique< unsigned char []>(
195
+ image.cols * image.rows * 3 * batch_size);
195
196
}
196
197
int index = b * image.cols * image.rows * 3 ;
197
198
// Copy into pinned memory
0 commit comments