- 深度学习图像识别技术:基于TensorFlow Object Detection API和OpenVINO? 工具套件
- 庄建 张晶 许钰雯编著
- 219字
- 2021-06-11 18:56:40
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文件夹内。