tensorflow api训练3（ckpt转成pb和pbtxt）完结

9. 将训练好的ckpt模型转成pb模型

我们只需要传入3个数据，一个是config路径，这里的config是我们训练时得到了，也可以用自己改的那个，ckpt路径和次数，输出的路径

python object_detection/export_inference_graph.py --input_type=image_tensor --pipeline_config_path=I:\persontrain\test3\pipeline.config --trained_checkpoint_prefix=I:\persontrain\test3\model.ckpt-25922 --output_directory=I:/persontrain/test3

回车之后就会生成pb模型，

10. 利用pb模型生成opencv dnn模块支持的pbtxt

进入下面的路径，然后cmd

python tf_text_graph_ssd.py --input I:\persontrain\test3\frozen_inference_graph.pb --output I:\persontrain\test3\graph.pbtxt --config I:\persontrain\test3\pipeline.config

11. python和c++都测试成功

tensorflow api训练3（ckpt转成pb和pbtxt）完结相关推荐

tensorflow实战：由ckpt转成pb步骤的总结
最近真的项目很紧,996模式开启了多日,在完成项目一小步后,总算有时间进行一些梳理,也是对tensorflow有了更多的认识. 之前在学校实验室中,其实并不涉及太多tensorflow c++端的接口 ...
动动手，用TensorFlow API训练出自己的目标检测模型
TensorFlow内包含了一个强大的物体检测API,我们可以利用这API来训练自己的数据集实现特殊的目标检测. Dat Tran就分享了自己实现可爱的浣熊检测器的经历,在文章中作者把检测器的训练流程 ...
使用tensorflow object detection API 训练自己的目标检测模型（三）
在上一篇博客"使用tensorflow object detection API 训练自己的目标检测模型 (二)"中介绍了如何使用LabelImg标记数据集,生成.xml文件,经过 ...
基于TensorFlow Object Detection API训练自己的目标识别模型
基于TensorFlow Object Detection API训练自己的目标识别模型环境 Windows10 CUDA_9 Cudnn_9.0 Anaconda3-5.2.0 Tensorflo ...
tflearn 中文汉字识别，训练后模型存为pb给TensorFlow使用——模型层次太深，或者太复杂训练时候都不会收敛...
tflearn 中文汉字识别,训练后模型存为pb给TensorFlow使用. 数据目录在data,data下放了汉字识别图片: data$ ls 0 1 10 11 12 13 14 1 ...
tensorflow object detection API训练错误解决
问题描述 tensorflow object detection API训练coco数据集时提示错误:Windows fatal exception: access violation,如下图: Th ...
关于使用tensorflow object detection API训练自己的模型-补充部分（代码，数据标注工具，训练数据，测试数据）
之前分享过关于tensorflow object detection API训练自己的模型的几篇博客,后面有人陆续碰到一些问题,问到了我解决方法.所以在这里补充点大家可能用到的东西.声明一下,本人专业 ...
使用tensorflow object detection API 训练自己的目标检测模型（二）labelImg的安装配置过程
上一篇博客介绍了goggle的tensorflow object detection API 的配置和使用, 这次介绍一下如何用这个API训练一个私人定制的目标检测模型. 第一步:准备自己的数据集.比 ...
Java / Tensorflow - API 调用 pb 模型使用 GPU 推理
目录一.引言二.Java / Tensorflow 代码配置 1.代码配置 2.Maven 配置三.环境检测 1.显卡检测 2.显卡监控四.推理踩坑 1.异常现象 2.异常日志五.安装 cu ...
tensorflow框架.ckpt .pb模型节点tensor_name打印及ckpt模型转.pb模型
转换模型首先要知道的是从哪个节点输出,如果没有源代码是很难清楚节点信息. 获取ckpt模型的节点名称 import os from tensorflow.python import pywrap_te ...

tensorflow api训练3（ckpt转成pb和pbtxt）完结

9. 将训练好的ckpt模型转成pb模型

10. 利用pb模型生成opencv dnn模块支持的pbtxt

11. python和c++都测试成功

tensorflow api训练3（ckpt转成pb和pbtxt）完结相关推荐

最新文章

热门文章