介绍
一种转换工具,可将YOLO v3暗网权重转换为TF Lite模型(YOLO v3 PyTorch> ONNX> TensorFlow> TF Lite)和TensorRT模型(dynamic_axes分支)。
先决条件
python3
torch==1.3.1
torchvision==0.4.2
onnx==1.6.0
onnx-tf==1.5.0
onnxruntime-gpu==1.0.0
tensorflow-gpu==1.15.0
码头工人
docker pull zldrobit/onnx:10.0-cudnn7-devel
用法
1.下载预训练的Darknet权重:
cd weights
wget https://pjreddie.com/media/files/yolov3.weights
2.将YOLO v3模型从Darknet权重转换为ONNX模
1