- 博客(3)
- 收藏
- 关注
原创 NMS加速
②分类概率 * 前景概率 < confidence_threshold时, 跳过。跳过① ②:587ms (可能测试图片box少,if判断用时大于计算用时)①前景概率 < confidence_threshold时,跳过。数学运算需要的时间远超过很多if判断。不需要计算每个BOX两两之间的IOU。都不跳过:2.6064e+06ms。跳过②:5053ms。
2024-03-02 15:26:55
387
原创 【Yolo 视频和摄像头推理】C++ TensorRT
在while循环中添加推理代码与图片推理有一些不同,如果把模型初始化定义和开辟空间的部分一同放进去,会造成一直循环重复多余工作导致运行速度下降,严重情况会一直开辟不同的空间造成电脑内存资源爆满。使用cv::VideoCapture的方法,使用视频则放入视频的地址;本文介绍了使用yolov5进行C++视频和摄像头的使用,难点在于循环处理每帧图片时尽量把不相关的部分放在循环外避免资源使用过多。由以下代码可以看出,我们只需把推理代码放入读取和写入之间,便完成整个视频的目标检测。更多细节基础部分请参考上篇写的。
2023-10-08 16:42:00
1345
4
原创 【详细案例】yoloV5 tensorRT加速 C++图片推理
@TOC前言因部署jetson nano需要,以yolov5为例提供一种tensorRT加速推理的实例。python训练出来的神经网络权重文件pt如果想要应用到实际设备上,需要进行权重文件的转换工作:pt -> onnx -> engine(TensorRT),加载engine文件可以更流畅的运行推理,这在预测模型移植到小型设备上是有必要的工作。项目文件已上传至github源码一、权重文件pt转onnxonnx是一种通用神经网络文件,使用这种文件格式可脱离pytorch、te
2023-10-06 14:32:51
3068
7
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人