项目场景:
c++中同时使用tensorRT与opencv
问题描述
使用tensorRT进行推理,使用opencv读取和处理图像。我定义了一个trt.hpp的头文件,并在里面包含了trt的头文件,为了把cv::Mat转为我自己封装的图像类Image
中,我定义了一个函数:
trt::Image cv2img(const std::pair<cv::Mat, int>& inp)
{
return trt::Image(inp.first.data, inp.first.cols, inp.first.rows, inp.second);
}
其实就是调用一下Image的构造函数,但是由于我把这个函数定义在trt中,导致opencv提示一大堆warning #611-D: overloaded virtual function "cv::" is only partially overrid。
解决方案:
解决方案应该蛮多的,比如直接屏蔽掉这个warning,应该也无伤大雅,或者把这个函数抽出来放其他地方,只要不在include tensorrt相关头文件的地方同时include opencv的头文件应该就不会有问题(include tensorrt的空间里,include的其他头文件里面也不能包含opencv)。
如果非要硬整,例如我这里就非要把cv2img放到trt.h中,可以采用模板函数:
template <typename T, typename D>
T cv2img(const std::pair<D, int>& inp)
{
return T(inp.first.data, inp.first.cols, inp.first.rows, inp.second);
}
这样在trt.h中就不需要包含opencv,只需要在调用该函数的位置包含opencv就行,效果是一样的,也不会触发warning,但是不一定适用所有人。