您好,
我使用由matterport的MASK-RCNN模型修改的版本(使用tensorflow 2.2.0), 生成了一个custom datasets的H5模型, 想要将h5转为cv2.dnn.readFromTensorflow可用的pb檔,
我看了网上许多生成pb的方法, 其中一个Frozen-Graph-Inference-MaskRCNN, 确实可行, 并增加其中tf.train.write_graph(frozen_graph, directory, name , as_text=False)的as_text=True即可同时生成pb, pbtxt檔, 但也无法在opencv中使用, 刚好看见您的文章, 真是喜出望外啊, 可执行了后, 又出现以下问题, 希望您能给我一个方向
非常谢谢您
AttributeError Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_25268/3656456214.py in
1 out_prefix = "output_"
----> 2 for i in range(len(keras_model.outputs)):
3 out_nodes.append(out_prefix + str(i + 1))
4 tf.identity(keras_model.output[i], out_prefix + str(i + 1))
5 model_name = os.path.splitext(os.path.split(model_path)[-1])[0] + '_' + mode + '.pb'
AttributeError: 'NoneType' object has no attribute 'outputs'