python list 转 label_map.txt

转换过程：

def convert_classes(classes_list, output_path,start_ID=1):
    end = '\n'
    s = ' '
    class_map = {}
    for ID, name in enumerate(classes_list):
        out = ''
        out += 'item' + s + '{' + end
        out += s * 2 + 'id:' + ' ' + (str(start_ID)) + end
        out += s * 2 + 'name:' + ' ' + '\'' + name + '\'' + end
        out += '}' + end * 2

        with open(output_path, 'a') as f:
            f.write(out)

        class_map[name] = start_ID
        start_ID +=1

if __name__ == '__main__':
    classes_list = ['cat', 'dog', 'car', 'person']
    output_path = './label_map.pbtxt'
    convert_classes(classes_list,output_path,start_ID=0)

结果：

item {
  id: 0
  name: 'cat'
}

item {
  id: 1
  name: 'dog'
}

item {
  id: 2
  name: 'car'
}

item {
  id: 3
  name: 'person'
}

参考