laugh12321/TensorRT-YOLO

[Question]: 关于【使用了 CUDA 核函数来加速前处理过程】在哪里体现

songsong695 opened this issue · 2 comments

songsong695 commented

您好，请问”使用了 CUDA 核函数来加速前处理过程“，体现在哪里呢，是在ImageBatcher类里面吗，没有发现加速的部分呢，谢谢！

laugh12321 commented

【CUDA核函数加速前处理】在老版本的Python代码中有使用，因为在Python上无法实现真正的多stream并行（batch是一不影响），就改为给C++用了。你要是需要后面可以加上。

  • 👍1
songsong695 commented

【CUDA核函数加速前处理】在老版本的Python代码中有使用，因为在Python上无法实现真正的多stream并行（batch是一不影响），就改为给C++用了。你要是需要后面可以加上。

好的，谢谢您！