上QQ阅读APP看书,第一时间看更新
3.5 依据标注类型创建标签映射文件
完成标注后,TensorFlow Object Detection API要求依据标注类型创建标签映射文件label_map.pbtxt,文件格式如代码清单3-1所示。
代码清单3-1 标签映射文件
每一个标注类型对应一个item项。每个item项有两个属性,id和name。
id的编号从1开始,若有两个类型,则从1开始到2结束;若有4个类型,则从1开始到4结束,以此类推。
name是显示给人看的字符串,若类型为cat,则name为'cat'。
根据上述规则,猫狗项目的标签映射文件label_map.pbtxt如代码清单3-2所示。
代码清单3-2 label_map.pbtxt
编辑完成后,请把label_map.pbtxt文件放到annotations文件夹内。