LabelImg软件制作tfrecord格式数据集

最新推荐文章于 2023-06-30 16:23:00 发布

Nie2014

最新推荐文章于 2023-06-30 16:23:00 发布

阅读量723

点赞数

分类专栏： tensorflow安装文章标签： tensorflow

本文链接：https://blog.csdn.net/Nie2014/article/details/107035454

版权

本文详细介绍了如何使用LabelImg软件进行图像标注，将XML文件转换为CSV，然后将CSV转换为TFRecord格式，适用于TensorFlow训练。内容包括LabelImg的下载安装、图像标注、XML到CSV的转换、配置TensorFlow环境、Protobuf编译以及生成TFRecord的步骤。

摘要由CSDN通过智能技术生成

下载资源

1、labelImg标注工具的下载及安装：
下载软件： tzutalin /labelImg
安装：参看作者的项目文档，找到Installation，选择window+anaconda的安装方式：
在这里插入图片描述

进行标记

1、打开labelimg软件
打开解压缩后的labelimg文件存放地址：
在这里插入图片描述运行labelImg.py程序

则打开labelImg软件界面：
2、设置导入、导出操作地址

将待标记图像集存放在data-picture文件夹中，选择opendir命令，打开data-picture中的数据集并完成标记。
打开change save dir，将导出地址选择为data-xml文件夹。如果未做修改保持默认，则会将.xml文件输出在data-picture文件夹中
在这里插入图片描述
3、标记图像
至此，完成了图片集的标记，软件其他的详细命令请参考手把手教你图片打标。
查看生成的.xml文件：因为选择了Change Save Dir命令，所以输出.xml在data-xml的train文件夹中：

在这里插入图片描述
４、YOLO数据集的标注格式的annotation

在labelImg的项目文档中如下说明：
在这里插入图片描述标注的生成格式：1、ImageNet数据集使用PASCAL VOC格式
2、YOLO使用的格式

生成的yolo标注格式：

VOC格式的.xml文件——>tfrecord

.xml 转csv

在这里插入图片描述生成的.csv文件内容如下：

其中，xml_to_csv.py程序内容，里面有2项内容需要根据自己的实际情况更改：

import os
import glob
import pandas as pd
import xml.etree.ElementTree as ET

def xml_to_csv(path):
    xml_list = []
    print(path+'1')
    a = glob.glob(path + '/*.xml')
    print(a)
    for xml_file in glob.glob(path + '/*.xml'):
        print(path + '2')
        print(path + '/*.xml')
        tree = ET.parse(xml_file)
        root = tree.getroot()
        for member in root.findall('object'):
            value = (root.find('filename').text,
                     int(root.find('size')[0].text),
                     int(root.find('size')[1].text),
                     member[0].text,
                     int(member[4][0].text),
                     int(member[4][1].text),
                     int(member[4][2]