
ONNX_TFLITE_YOLOv3: 用于将YOLO v3 Darknet权重转为TF Lite模型的转换工具(基于PyTorch)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
ONNX_TFLITE_YOLOv3是一款专为开发者设计的工具,能够高效地将YOLO v3 Darknet格式的权重文件转换成TensorFlow Lite模型,支持PyTorch框架。
介绍一种转换工具,可以将YOLO v3的Darknet权重文件转换为TF Lite模型(路径:YOLO v3 PyTorch > ONNX > TensorFlow > TF Lite)以及TensorRT模型(dynamic_axes分支)。使用此工具前,请确保安装以下依赖项:
- python 3
- 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进行环境搭建:
```shell
docker pull zldrobit/onnx:10.0-cudnn7-devel
```
转换步骤如下:
1. 下载预训练的Darknet权重文件。
- 进入weight目录:`cd weights`
- 使用wget命令下载权重文件:`wget https://pjreddie.com/media/files/yolov3.weights`
2. 将YOLO v3模型从Darknet格式转换为ONNX格式。
全部评论 (0)
还没有任何评论哟~


