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')

我收到失?。?/p>

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。