search

bubbliiiing / yolov4-pytorch

这是一个YoloV4-pytorch的源码，可以用于训练自己的模型。
MIT License
2.08k stars 612 forks source link

opencv加载图像进行推理 #361

Open PengPeng-JunJun opened 1 year ago

PengPeng-JunJun commented 1 year ago

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

bubbliiiing commented 1 year ago

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

PengPeng-JunJun commented 1 year ago

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

PengPeng-JunJun commented 1 year ago

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

bubbliiiing commented 1 year ago

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

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

bubbliiiing commented 1 year ago

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

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