1

使用迁移学习,我在 TensorFlow (tensorflow-gpu==1.15.0) 中训练了 SSD MobileNetV2 (ssd_mobilenet_v2_coco.config) 模型。在使用 TensorFlow API Python 脚本 (export_inference_graph.py) 冻结图 (.pb) 后,我使用 OpenCV wiki (tf_text_graph_ssd.py) 中提供的 Python 脚本创建了一个文本图 (.pbtxt)。

我使用 wiki 中的 Python 代码片段来测试推理,但出现以下错误:

cv2.error: OpenCV(4.2.0) C:\projects\opencv-python\opencv\modules\dnn\src\dnn.cpp:562: error: (-2:Unspecified error) Can't create layer "FeatureExtractor/MobilenetV2/expanded_conv_2/add" of type "AddV2" in function 'cv::dnn::dnn4_v20191202::LayerData::getLayerInstance'

我正在使用 Windows 10、Python 3.6.8 和 OpenCV 4.2.0.32。我曾尝试降级 OpenCV,但早期版本会给出不同的错误。

但是,在 Ubuntu 18.04.4 中,从源代码安装 OpenCV,我没有收到任何错误。有人知道这是否是 Windows 的 OpenCV 二进制轮子中不兼容的层吗?我应该等到下一个版本吗?

4

0 回答 0