【Tensorflow】Darknet的yolov3.weights文件转换成tensorflow的ckpt或者pb文件

原文地址：https://blog.csdn.net/heiheiya/article/details/91437196

感谢 heiheiya 。

做个学习笔记

一、下载源码

git clone https://github.com/mystic123/tensorflow-yolo-v3.git
cd tensorflow-yolo-v3

　

1. 转换成ckpt文件格式

　YOLOv3：

python convert_weights.py --class_names coco.names --data_format NHWC --weights_file yolov3.weights

　默认在当前文件夹下新建一个saved_model文件夹，里面是转换生成的文件

2. 转换成pb文件格式

YOLOv3：

python convert_weights_pb.py --class_names coco.names --data_format NHWC --weights_file yolov3.weights

默认在当前文件夹下生成一个pb文件

YOLOv3-tiny：　　

python convert_weights_pb.py --class_names coco.names --data_format NHWC --weights_file yolov3-tiny.weights --tiny

　如果是转换自己训练的数据集，则将coco.names和yolov3.weights替换成自己相应的文件就可以了。