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

 

相关文章:

  • 2021-10-22
  • 2022-12-23
  • 2021-10-15
  • 2022-12-23
  • 2022-12-23
  • 2019-11-28
  • 2022-12-23
  • 2021-12-27
猜你喜欢
  • 2022-02-01
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-09-18
  • 2021-04-10
相关资源
相似解决方案