簡體   English   中英

使用tensorflow時,在哪里可以找到ssd_mobilenet_v1_coco的標簽映射文件(pbtxt)?

[英]Where can I find the label map file(pbtxt) of ssd_mobilenet_v1_coco when using tensorflow?

我使用tensorflow object detection在 windows 10 上學習對象檢測

我從https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md下載 ssd_mobilenet_v1_coco_2018_01_28.tar.gz

我解壓ssd_mobilenet_v1_coco_2018_01_28.tar.gz文件后,沒有找到pbtxt文件。

ssd_mobilenet_v1_coco的相關pbtxt文件在哪里可以找到?

我知道在models-master\\research\\object_detection\\data文件夾中有一些 pbtxt 文件,但是哪個文件與 ssd_mobilenet_v1_coco 相關?

標簽圖並不特定於架構,而是特定於數據集(您支持哪些類,因此您應該設置分類層的通道數)。 因此,您只需要與 coco 對應的標簽圖,即object_detection/data/mscoco_label_map.pbtxt

數據集在模型中實現。 模型由 tar.gz 或 .zip 提供。 如果您使用預訓練的基本模型,則可以在代碼樹 githup object_detection/data/mscoco_label_map.pbtxt 中找到標簽圖,如 netanel-sam 解釋的那樣。

但是,如果您開始訓練您的預訓練模型並將要檢測的項目添加到您的數據集中並開始交付您修改后的模型,那么您還必須提供您的標簽地圖,並且沒有比將其包含到 .tar.gz 或。壓縮。 lite-model 的情況相同,因為從可訓練模型到 lite 的轉換通常會丟失數據集中的項目。 Lite-model 還使用其他格式來標記地圖而不是基本模型。 令人困惑?

暫無
暫無

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

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