谷歌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就可以了，它会像魅力一样工作。