使用Opencv在基于SSD-MobileNet迁移学习中生成pbtxt文件遇到的问题及解决方案

在迁移学习（使用MobileNet）训练后，如果要是用opencv调用pb模型，需要的“佐料”是pbtxt文件，这个文件必不可少。
我在使用opencv4.0.0中的tf_text_graph_ssd.py（文件路径在“D:\opencv\opencv_4.0.0\sources\samples\dnn”）生成pbtxt文件时，遇到了这样一个问题：

Traceback (most recent call last):File "tf_text_graph_ssd.py", line 276, in <module>createSSDGraph(args.input, args.config, args.output)File "tf_text_graph_ssd.py", line 129, in createSSDGraphassert(graph_def.node[0].op == 'Placeholder')
AssertionError

如下图：

这时候的解决方案则是把frozen_inference_graph.pb转换一下，使用python程序：

import tensorflow as tf
from tensorflow.tools.graph_transforms import TransformGraphwith tf.gfile.FastGFile('F:/live/face_data/pb_model/frozen_inference_graph.pb', 'rb') as f:graph_def = tf.GraphDef()graph_def.ParseFromString(f.read())graph_def = TransformGraph(graph_def, ['image_tensor'], ['detection_boxes', 'detection_classes', 'detection_scores', 'num_detections'], ['sort_by_execution_order'])with tf.gfile.FastGFile('F:/live/face_data/pb_model/sorted_inference_graph.pb', 'wb') as f:f.write(graph_def.SerializeToString())

然后重新调用cmd命令进行转化即可。
结果如图：

使用Opencv在基于SSD-MobileNet迁移学习中生成pbtxt文件遇到的问题及解决方案相关推荐

平潭迁移库是什么意思_迁移学习》第四章总结---基于模型的迁移学习
基于模型的迁移学习可以简单理解为就是基于模型参数的迁移学习,如何使我们构建的模型可以学习到域之间的通用知识. 1. 基于共享模型成分的迁移学习在模型中添加先验知识. 1.1 利用高斯过程的迁移学习 ...
keras冻结_【连载】深度学习第22讲：搭建一个基于keras的迁移学习花朵识别系统（附数据）...
在上一讲中,和大家探讨了迁移学习的基本原理,并利用 keras 基于 VGG16 预训练模型简单了在 mnist 数据集上做了演示.鉴于大家对于迁移学习的兴趣,本节将继续基于迁移学习利用一些花朵数据搭 ...
基于 CNN 和迁移学习的农作物病害识别方法研究
基于 CNN 和迁移学习的农作物病害识别方法研究 1.研究思路采用互联网公开的 ImageNet 图像大数据集和PlantVillage 植物病害公共数据集, 以实验室的黄瓜和水稻病害数据集 AES ...
机器学习工程师 — Udacity 基于CNN和迁移学习创建狗品种分类器
卷积神经网络(Convolutional Neural Network, CNN) 项目:实现一个狗品种识别算法App 推荐你阅读以下材料来加深对 CNN和Transfer Learning的理解: ...
迁移学习中的负迁移：综述
点击上面"脑机接口社区"关注我们更多技术干货第一时间送达导读迁移移学习(TL)试图利用来自一个或多个源域的数据或知识来促进目标域的学习.由于标记成本.隐私问题等原因,当目标域 ...
在基于图像的深度学习中如何做数据的自动标注以及自动标注的等级介绍
作者:Tobias Schaffrath Rosario 编译:ronghuaiyang 原文:在基于图像的深度学习中如何做数据的自动标注以及自动标注的等级介绍_ronghuaiyang的博客-CSD ...
迁移学习中使用下载好的权重
迁移学习中使用下载好的权重 1.最开始使用Vgg16 2.将下载好的权重放到对应的目录将下载好的权重放到./keras/models中改用其他内嵌网络会发现,直接可以用了
ML.NET 示例：图像分类模型训练-首选API（基于原生TensorFlow迁移学习）
ML.NET 版本 API 类型状态应用程序类型数据类型场景机器学习任务算法 Microsoft.ML 1.5.0 动态API 最新控制台应用程序和Web应用程序图片文件图像分类基 ...
Nat. Mach. Intell. | 基于神经网络的迁移学习用于单细胞RNA-seq分析中的聚类和细胞类型分类...
今天给大家介绍由美国宾夕法尼亚大学佩雷尔曼医学院生物统计学,流行病学和信息学系Jian Hu等人在<Nature Machine Intelligence>上发表了一篇名为"It ...
竞赛获奖系统解读：远场说话人确认中基于两阶段迁移学习解决域不匹配问题
作为Interspeech2022的赛事活动,远场说话人验证挑战赛 (FFSVC) 由昆山杜克大学.新加坡国立大学.南加州大学和希尔贝壳联合组织,主要关注极具挑战性的远场说话人确认任务.2020年举办 ...

使用Opencv在基于SSD-MobileNet迁移学习中生成pbtxt文件遇到的问题及解决方案

使用Opencv在基于SSD-MobileNet迁移学习中生成pbtxt文件遇到的问题及解决方案相关推荐

最新文章

热门文章