使用 OpenCV readNetFromTensorFlow 运行 Keras DNN

Answer 1

Python 版本 3.7

凯拉斯版本 2.3.1

TensorFlow 版本 1.14.0

我想在 C++ 中使用 OpenCV 的 readNetFromTensorflow 运行我的 UNet Keras 模型。根据这个问题，我已经成功地将我的 HDF5 文件转换为 .pb： How we can convert keras model .h5 file to tensorflow saved model (.pb)

但是，当我尝试运行命令时（为了便于测试，首先在 python 中）：

net = cv.dnn.readNetFromTensorflow('tensorflow/my_model.pb')

我收到失败：

error: (-2) Unknown layer type Shape in op decoder_stage0_upsampling/Shape in function cv::dnn::experimental_dnn_v2::`anonymous-namespace'::TFImporter::populateNet

使用 OpenCV 有解决方法吗？或者在这种情况下最好使用 Tensorflows C++ API。