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


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

感谢  。

做个学习笔记

一、下载源码

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替换成自己相应的文件就可以了。 

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM