Darknet yolov4 转onnx

前段时间,自己瞎搞,需要把Darknet yolov4转成onnx格式,在此记录一下过程。
我们需要在github上下载以为大佬写好的转换代码

https://github.com/Tianxiaomo/pytorch-YOLOv4

执行如下命令:

cd pytorch-YOLOv4
python3 demo_darknet2onnx.py ./yolov4.cfg ./data/coco.names ./yolov4.weights ./data/dog.jpg 1

然后会得到onnx
Darknet yolov4 转onnx_第1张图片
并且会预测一张图片并保存在当前路径下
Darknet yolov4 转onnx_第2张图片
Darknet yolov4 转onnx_第3张图片

你可能感兴趣的:(darknet,yolov4,onnx,pytorch,深度学习,计算机视觉)