Skip to content

Commit 218ab07

Browse files
author
lyu
committed
update pipeline
1 parent 4b88bda commit 218ab07

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

BEVDet-TensorRT.png

-487 KB
Loading

src/preprocess.cu

+2-2
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ int preprocess(const uchar* src_imgs, float* dst_imgs, int n_img, int src_img_h,
199199
dim3 grid(dst_img_h, dst_img_w);
200200
dim3 block;
201201
if(sample == Sampler::bicubic){
202-
printf("sampler : bicubic\n");
202+
// printf("sampler : bicubic\n");
203203
block = dim3(n_img, 16);
204204
fill_in_kernel<<<dim3(dst_img_h, dst_img_w), dim3(n_img * 3)>>>(dst_imgs, 0.0f);
205205

@@ -210,7 +210,7 @@ int preprocess(const uchar* src_imgs, float* dst_imgs, int n_img, int src_img_h,
210210
offset_w, mean, std);
211211
}
212212
else if(sample == Sampler::nearest){
213-
printf("sampler : nearest\n");
213+
// printf("sampler : nearest\n");
214214
block = dim3(n_img);
215215
preprocess_nearest_kernel<<<grid, block>>>(src_imgs, dst_imgs, src_row_step, dst_row_step,
216216
src_img_step, dst_img_step, src_img_h, src_img_w, resize_radio_h,

0 commit comments

Comments
 (0)