OpenCV中的Tensorflow自定義模型

Question

我已經在ssd_mobilenet_v1_coco上訓練了一個新模型，用於自定義數據集。 該模型在tensorflow中工作良好。 但是現在我想在OpenCV中使用它。

net = cv2.dnn.readNetFromTensorflow("model/frozen_inference_graph.pb", "model/protobuf.pbtxt")
detections = net.forward()

因此，對於配置文件，我將Frozen_graph轉換為pbtxt並將其添加。 但是隨后出現以下錯誤

[libprotobuf ERROR /home/chamath/Projects/opencv/opencv/3rdparty/protobuf/src/google/protobuf/text_format.cc:298] Error parsing text-format tensorflow.GraphDef: 731:5: Unknown enumeration value of "DT_RESOURCE" for field "type".

如此處的建議，我嘗試使用線程中提到的此配置文件，但是當我使用它時，對象檢測無法正常工作。 檢測到不正確的平方數，並且將它們放錯了位置。

有什么方法可以創建與OpenCV兼容的pbtxt配置文件？ 或關於如何使我的模型在OpenCV中工作的任何建議？

Answer 1

經過培訓，您可能還沒有生成適當的圖形。 您必須像這樣轉換圖形：

python ../opencv/samples/dnn/tf_text_graph_ssd.py-輸入接受訓練的推理圖/inference_graph_v5.pb/frozen_inference_graph.pb-輸出接受訓練的推理圖/inference_graph_v5.pb/graph.pbtxt

然后通過。 pb和graph.pbtxt到DNN.readNetFromTensorflow應該適合您的:)