[英]Avoid overlapping bounding boxes in Tensorflow Object Detection 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。 有沒有辦法做到這一點?
我從訓練中生成了三個文件:
預測框是否包含在這些文件之一中? 或者它們存儲在其他地方? 或者它們是否需要單獨編碼才能提取坐標?
您列出的文件是檢查點文件,您可以使用 then 導出凍結圖,然后對輸入圖像進行預測。
獲得凍結圖后,您可以使用此文件object_detection_tutorial.ipynb
對輸入圖像進行預測。 在此文件中,函數run_inference_for_single_image將為每個圖像返回一個輸出字典,其中包含檢測框。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.