yolov4-pytorch icon indicating copy to clipboard operation
yolov4-pytorch copied to clipboard

opencv加载图像进行推理

Open PengPeng-JunJun opened this issue 1 year ago • 5 comments

代码中的推理和预测都是使用的PIL库加载的图像,请问有测试过用opencv加载图像进行推理么? 预测中有推理vedio的代码,但是将代码复制到推理单张图像时,运行会报错,在utils.py文件的20行,个人认为是cv和PIL的API不一致的原因

PengPeng-JunJun avatar Apr 07 '23 09:04 PengPeng-JunJun

我使用起来都差不多,使用PIL是因为绘图

bubbliiiing avatar Apr 07 '23 15:04 bubbliiiing

可以分享一份用OPENCV来实现推理预测的代码么?当前的代码都是用PIL来加载图像的

PengPeng-JunJun avatar Apr 09 '23 14:04 PengPeng-JunJun

up主,还想请教一下,代码中有将pth文件转换成ONNX的功能,请问你有用过ONNX来进行过推理预测么?是用的OPENCV的DNN模块或是ONNXRuntime还是NCNN这一类的推理加速引擎呢?

PengPeng-JunJun avatar Apr 09 '23 14:04 PengPeng-JunJun

可以分享一份用OPENCV来实现推理预测的代码么?当前的代码都是用PIL来加载图像的

这个我不考虑,因为这两者没有本质区别,性能差距也基本没有……

bubbliiiing avatar Apr 09 '23 15:04 bubbliiiing

up主,还想请教一下,代码中有将pth文件转换成ONNX的功能,请问你有用过ONNX来进行过推理预测么?是用的OPENCV的DNN模块或是ONNXRuntime还是NCNN这一类的推理加速引擎呢?

里面有用onnx读取然后预测的代码

bubbliiiing avatar Apr 09 '23 15:04 bubbliiiing