如何使用openCV读取神经网络模型

Question

我正在使用cv2.dnn.readNet将神经网络模型用于图像中的文本检测。它会出现以下错误，我不知道它在说什么。

这是我的代码片段：

nn = cv2.dnn.readNet('frozen_east_text_detection.pb')

错误是：

error: OpenCV(4.1.2) /io/opencv/modules/dnn/src/tensorflow/tf_io.cpp:42: error: (-2:Unspecified error) FAILED: ReadProtoFromBinaryFile(param_file, param). Failed to parse GraphDef file: frozen_east_text_detection.pb in function 'ReadTFNetParamsFromBinaryFileOrDie'

我应该说我在google colab上写作，谢谢...

Answer 1

我要尝试的第一件事是确保您的路径正确。 frozen_east_text_detection.pb是否与python脚本位于同一文件夹或位置？尝试将完整路径添加到函数中，同时还要在路径的开头添加r，以告知python您没有在使用转义字符。如果是这样，我也将您的第一句话改为以下内容。

nn = cv2.dnn.readNet("frozen_east_text_detection.pb")

但是，如果仍然不能解决问题，我想您正在使用的文件已在某种程度上损坏了，建议您尝试寻找另一个下载该文件的位置。根据我的研究，该文件是项目或教程的一部分，因此找到另一个副本并不难。