|
@@ -64,6 +64,7 @@ void DepthModelImpl::preprocess(const tensor::Image &image, affine::LetterBoxMat
|
|
|
image.height, input_device, network_input_width_,
|
|
|
network_input_height_, affine_matrix_device, 114,
|
|
|
normalize_, stream_);
|
|
|
+ checkRuntime(cudaStreamSynchronize(stream_));
|
|
|
}
|
|
|
|
|
|
|
|
@@ -81,6 +82,7 @@ void DepthModelImpl::postprocess(int width, int height, void *stream)
|
|
|
image_device, network_input_width_, network_input_width_, network_input_height_,
|
|
|
dst_device, width, height, invert_affine_matrix_device, 0,
|
|
|
stream_);
|
|
|
+ checkRuntime(cudaStreamSynchronize(stream_));
|
|
|
}
|
|
|
|
|
|
Result DepthModelImpl::forward(const tensor::Image &image, int slice_width, int slice_height, float overlap_width_ratio, float overlap_height_ratio, void *stream)
|