谷歌tensorflow objectdetection培訓

Question

我按照說明在這裏：
https://github.com/tensorflow/models/blob/master/object_detection/g3doc/running_pets.md

當我轉換到了這一步：

python object_detection/create_pet_tf_record.py \ 
    --label_map_path=object_detection/data/pet_label_map.pbtxt \ 
    --data_dir=`pwd` \ 
    --output_dir=`pwd` 

發生這種情況：

File "/work/tensorflow/models/object_detection/utils/label_map_util.py", line 36, in _validate_label_map 
    raise ValueError('Label map ids should be >= 1.') 
ValueError: Label map ids should be >= 1. 

我該如何解決？

Answer 1

您的圖片ID可能從0開始。 api將0視爲背景並因此引發值錯誤。只需增加1就可以了，它會像魅力一樣工作。