win10系统下使用labelimg 标记自己的数据集并生成tfrecord文件并开始训练 ,将模型复用，用于识别

xifan021

于 2018-09-10 18:49:56 发布

阅读量4.6k

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/xifan021/article/details/82427402

本文档详细介绍了如何在Windows 10上使用LabelImg工具标注数据集，生成TFRecord文件，并基于TensorFlow Object Detection API训练模型。首先从GitHub下载并安装LabelImg，然后创建文件夹结构，重命名图片并进行标注。接着，自定义创建_pascal_tf_record.py脚本，编辑路径和内容，生成train和val记录文件。在训练过程中，遇到了模型要求3通道输入的问题，通过OpenCV将灰度图转换为3通道解决。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 从github下载labelimg:https://github.com/tzutalin/labelImg#macos

2在D盘新建一个my_voc文件夹，在该文件夹下新建三个文件夹分名命名为：Annatations，ImageSets，JPEGImages其中JPEGImages中放入要标记的图片，修改jpg格式的图片的名称为000001格式：（*下载Bulk Rename Utility 余飞雨汉化版），先使用D:\f\rename.py重命名所有文件，再使用Bulk Rename Utility在修改后的文件名前面加0。Annatations用来存放标记后生成的xml文件。

3 打开labelimg.py运行，即可出现标记窗口

参考：https://blog.youkuaiyun.com/chenmaolin88/article/details/79357502

先点击open dir 修改文件路径为：D:\my_voc\JPEGImages

点击change save dir 修改xml文件的保存路径为：D:\my_voc\Annatations

4参考：https://blog.youkuaiyun.com/chenmaolin88/article/details/79357263

将D:\models-master\research\object_detection\dataset_tools\create_pascal_tf_record.py复制出来一个（还复制到同一个文件夹下），修改成create_pascal_tf_record_crack.py,修改其中内容成现在的样子

5在my_voc文件夹下新建文本文件：crack_label_map.pbtxt

输入：

item {
id: 1
name: 'crack'
}

新建train.txt和val.txt

利