将tf.keras vgg16分类器从.pb转换为pbtxt以在opencv dnn模块中使用

时间:2019-08-10 05:13:46

标签: opencv tensorflow keras tf.keras

我使用tf.keras优化了一个简单的vgg16模型,并使用

将其转换为.pb
new_model = tf.keras.experimental.load_from_saved_model('path_to_saved_model')

现在我想在opencv dnn模块中使用该模型,所以我需要一个pbtxt文件,我尝试使用一些脚本进行转换但失败了,所以我还读到您有时可以不用指定.pbtxt文件就可以摆脱困境,例如在东部文本中检测器,所以我尝试使用cv2.readfromnet(frozen.pb)方法,但仍然显示错误,并说在解析pbtxt文件时遇到错误

所以我尝试转换为onnx格式,因为它只是一个文件,但也遇到错误。

现在我的问题是,如果您已经成功地训练了自定义模型并将冻结的图转换为pbtxt,那么您将使用哪种方法(注意:我不是在训练对象检测器,而是在简单的vgg16分类器上训练),所以tensorflow中的脚本模范动物园对我没用。

已经在此花费了几个小时,因此,如果您有任何建议,不妨听听它

0 个答案:

没有答案