简介
该用户还未填写简介
擅长的技术栈
可提供的服务
暂无可提供的服务
【代码】yolo格式的txt与labelimg的xml文件互转。
【代码】OpenMMlab导出FCN模型并用onnxruntime推理。
方法一:通过trtexec转换onnx文件,LZ的版本是TensorRT-8.2.1.8。则三个输出分别为boxes、scores、class_ids。方法二:通过mmdeploy导出engine文件。
【代码】OpenMMlab导出mobilenet-v2的onnx模型并推理。
注意,由于C++无法直接读取npy格式文件(可以依赖一些库),这里先使用python脚本将npy文件转换成txt文件。以分16类50部分,gpu版本为例。以分10类,gpu版本为例。以分13类,gpu版本为例。
【代码】onnxruntime和tensorrt多batch推理。
【代码】OpenMMlab导出DETR模型并用onnxruntime推理。
这里通过trtexec转换onnx文件,LZ的版本是TensorRT-8.2.1.8。输出是包含多个检测头的输出。安装mmdeploy的话,可以通过下面脚本导出onnx模型。
python工程实现:(采用yolov5作为检测器,onnx模型推理)SORT 是一种简单的在线实时多目标跟踪算法。(1)以 IoU 作为前后帧间目标关系度量指标;(3)通过匈牙利算法关联检测框到目标;(2)利用卡尔曼滤波器预测当前位置;(4)应用试探期甄别虚检。
下面的代码搭建了简单的一个生产者-消费者模型,在capture()函数中进行入队操作,infer()函数中进行出队操作,为了模拟采图-推理流程,在函数中调用Sleep()函数延时。此时发现采图-推理流程不能同步。上面的程序还有一点小问题:视频播放完时程序无法正常退出。