多张图片预测:
https://github.com/noahmr/yolov5-tensorrt
https://github.com/noahmr/yolov5-tensorrt/blob/main/src/yolov5_detector.cpp
Result Detector::detectBatch(const std::vectorcv::cuda::GpuMat& images,
std::vector<std::vector<Detection>>* out,
int flags) noexcept
下载后:
yolov5-tensorrt-main.zip
跟上面基本一样:
cv::Mat trtInput = blobFromDsImages(vec_ds_images, _p_net->getInputH(),_p_net->getInputW());
yolov5-v6:
下载后:yolo-tensorrt-TRT8.zip
多图片预测2:
完整调试博客:
yolov5转tensorrt c++_jacke121的专栏-CSDN博客
https://github.com/OpenJetson/tensorrt-yolov5/blob/main/yolov5.cpp
for循环添加数据
static float data[BATCH_SIZE * 3 * INPUT_H * INPUT_W];
cudaMemcpy 效率不高:
https://github.com/Wulingtian/yolov5_tensorrt_int8/blob/master/yolov5s_infer.cc
版权说明 : 本文为转载文章, 版权归原作者所有 版权申明
原文链接 : https://blog.csdn.net/jacke121/article/details/121851414
内容来源于网络,如有侵权,请联系作者删除!