python - 如何使用 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 写作，谢谢你...

标签： pythonopencvneural-networkgoogle-colaboratory

Answer 1

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

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

但是，如果这仍然不起作用，我的猜测是您正在使用的文件在某种程度上已损坏，并且建议您尝试找到另一个位置来下载该文件。根据我的研究，该文件是项目或教程的一部分，因此找到另一个副本应该不难。