簡體   English   中英

輸出在 TensorFlow 對象檢測 API 上訓練的預測邊界框

[英]Output predicted bounding boxes trained on TensorFlow Object Detection API

我使用 Google TensorFlow 對象檢測 API [ https://github.com/tensorflow/models][1]使用 Faster RCNN inception v2 模型在我自己的數據集上進行訓練,並在 python 3 中編寫了一些我自己的腳本。它有效在我的視頻上相當不錯,現在我想輸出預測的邊界框來計算 mAP。 有沒有辦法做到這一點?

我從訓練中生成了三個文件:

  1. model.ckpt-6839.data-00000-of-00001
  2. model.ckpt-6839.index
  3. model.ckpt-6839.meta

預測框是否包含在這些文件之一中? 或者它們存儲在其他地方? 或者它們是否需要單獨編碼才能提取坐標?

您列出的文件是檢查點文件,您可以使用 then 導出凍結圖,然后對輸入圖像進行預測。

獲得凍結圖后,您可以使用此文件object_detection_tutorial.ipynb對輸入圖像進行預測。 在此文件中,函數run_inference_for_single_image將為每個圖像返回一個輸出字典,其中包含檢測框。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM