Opencv
文章平均质量分 58
永力
这个作者很懒,什么都没留下…
展开
-
opencv->tensorrt的数据排列方式
在使用tensorrt的时候,将opencv的Mat格式处理成tensorrt输入方式。中间还有图像预处理,包括通道的变换、图像尺寸变形,最重要的是如何将Mat格式处理成tensorrt 可接受的输入格式。Mat的图像数据,每个像素的三个通道是相邻存储的,因此要将Mat通道分离;使用split函数分离通道std::vector<float> prepareImage(std::vector<cv::Mat> &vec_img) { std::vector&l原创 2021-11-17 15:08:01 · 2963 阅读 · 1 评论 -
【opencv】pytorch->onnx->opencv模型推理
opencv的DNN推理模块是不直接支持pytorch训练出来的pth文件。一般来说,可以将pth文件转成onnx格式,再用opencv调用onnx文件,即可实现推理。pth文件转onnx文件时,因为onnx对于pth中某些定义的层是不支持的,难免会碰到一些问题。近期,对一些分割网络进行了部署,碰到的一些问题记录在下。RuntimeError: ONNX export failed: Couldn’t export operator aten::adaptive_avg_pool2donnx不支持原创 2021-08-28 17:40:43 · 3832 阅读 · 0 评论