tensorflow - 如何使用 python 编写此 pbtxt？

Question

如何使用 python 生成 label_map.pbtxt？我需要通过发送对象名称自动生成 label_map.pbtxt 文件。

item {
  id: 1
  name: 'object name'
}

我尝试使用 python 功能创建它，但是当我开始训练模型时出现此错误:

TypeError: a bytes-like object is required, not 'str'!

Answer 1

def label_map_v1(objname):
with open('/TEST-DS-TO-RECORD/annotations/label_map.pbtxt', 'a') as the_file:
    the_file.write('item\n')
    the_file.write('{\n')
    the_file.write('id :{}'.format(int(1)))
    the_file.write('\n')
    the_file.write("name :'{0}'".format(str(objname)))
    the_file.write('\n')
    the_file.write('}\n')

这可以生成我需要的文件。