Skip to content

Commit 0ded180

Browse files
committed
fix(tensorrt_yolox): fix cstyleCast to reinterpret_cast
Signed-off-by: Koichi Imai <kotty.0704@gmail.com>
1 parent 71fc4c3 commit 0ded180

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

perception/tensorrt_yolox/src/tensorrt_yolox.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -1205,7 +1205,7 @@ cv::Mat TrtYoloX::getMaskImageGpu(float * d_prob, nvinfer1::Dims dims, int out_w
12051205
cv::Mat mask = cv::Mat::zeros(out_h, out_w, CV_8UC1);
12061206
int index = b * out_w * out_h;
12071207
argmax_gpu(
1208-
(unsigned char *)argmax_buf_d_.get() + index, d_prob, out_w, out_h, width, height, classes, 1,
1208+
reinterpret_cast<unsigned char*>argmax_buf_d_.get() + index, d_prob, out_w, out_h, width, height, classes, 1,
12091209
*stream_);
12101210
CHECK_CUDA_ERROR(cudaMemcpyAsync(
12111211
argmax_buf_h_.get(), argmax_buf_d_.get(), sizeof(unsigned char) * 1 * out_w * out_h,

0 commit comments

Comments
 (0)