用opencv的dnn模块做yolov5目标检测的程序,包含两个步骤:1)、把pytorch的训练模型pth文件转换到onnx文件;2)、opencv的dnn模块读取onnx文件做前向计算。 1)、把pytorch的训练模型pth文件转换到onnx文件 yolov5官方代码:https://github.com/ultralytics/yolov5 这套程序里的代码
但是,从我的实验上来看,不管是CUDA 8.0 + cuDNN 5/6 或 CUDA 9.1 + cuDNN 7.1都是有问题的,可能的原因是GCC版本或者cuda本身问题,与caffe2本身关系不大。通过查看NVIDIA论坛上的讨论,最终使用CUDA 9.2 + cuDNN 7.1验证成功了caffe2源代码编译安装,CUDA 9.2 + cuDNN 7.1的安装方法可参考我在另一篇文章ubun...
用opencv的dnn模块做yolov5目标检测的程序,包含两个步骤:1)、把pytorch的训练模型pth文件转换到onnx文件;2)、opencv的dnn模块读取onnx文件做前向计算。 1)、把pytorch的训练模型pth文件转换到onnx文件 yolov5官方代码:https://github.com/ultralytics/yolov5 这套程序里的代码比较乱,在pytorch里,通常是在py文件...