python - 无法在python中将xml文件转换为csv

Question

为了使用 tensorflow 检测 API，我需要将 xml 文件转换为 csv 标签。我使用了这段代码，取自 datitran 的浣熊检测模型。他的github上有这个，你试试吧。 https://github.com/datitran/raccoon_dataset

但是，当我使用他的 xml_to_csv 代码时，我可以编译它，但新的标签文件是空的。这是代码，

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


def xml_to_csv(path):
xml_list = []
for xml_file in glob.glob(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].text),
                 int(member[4][3].text)
                 )
        xml_list.append(value)
column_name = ['filename', 'width', 'height', 'class', 'xmin', 'ymin', 'xmax', 'ymax']
xml_df = pd.DataFrame(xml_list, columns=column_name)
return xml_df


def main():
for directory in ['train','test']:
    image_path = os.path.join(os.getcwd(), 'images/{}'.format(directory))
    xml_df = xml_to_csv(image_path)
    xml_df.to_csv('data/{}_labels.csv'.format(directory), index=None)
    print('Successfully converted xml to csv.')


main()

其中“train”和“test”是包含 xml 文件的文件夹，images 是包含它们的文件夹，data 是我希望保存标签文件的文件夹。

Answer 1

我已使用上面给出的链接下载 github 存储库并查看了代码。首先，我复制上面的代码并粘贴到一个新文件中并运行它。

在上面发布的代码中，您获得空文件的原因是您为要调用的方法设置的路径在存储库中不存在，

如果我假设您想要使用 annotations 文件夹中提供的 xml 文件来创建新的 csv 文件来包含测试和单独训练数据，您所要做的就是在 Images 目录中创建两个名为 test 和 train 的新目录，然后复制 >xml 文件从 annotations 到您创建的这些新目录，然后运行上面的代码，它就可以完美运行

我只是做了同样的事情，只从 annotations 目录复制了一些 xml 文件当我这样做时，打印数据框以显示

 F:\expts\raccoon_dataset-master\raccoon_dataset-master>python import.py
          filename  width  height    class  xmin  ymin  xmax  ymax
0    raccoon-1.png    650     417  raccoon    81    88   522   408
1   raccoon-10.png    450     495  raccoon   130     2   446   488
2   raccoon-11.png    660     432  raccoon     3     1   461   431
3   raccoon-12.png    259     194  raccoon    28    21   126   181
4   raccoon-12.png    259     194  raccoon    85    33   235   193
5   raccoon-13.png    660     495  raccoon    55    28   393   313
6   raccoon-14.png    900     484  raccoon   163    81   546   438
7   raccoon-15.png    640     360  raccoon   313    61   614   360
8    raccoon-2.png    800     573  raccoon    60    51   462   499
9    raccoon-3.png    720     480  raccoon     1     1   720   476
10   raccoon-4.png    275     183  raccoon    21    11   200   183
11   raccoon-5.png    270     187  raccoon     3     3   260   179
12   raccoon-6.png    480     360  raccoon     1    44   307   316
13   raccoon-7.png    410     308  raccoon    92    79   271   264
14   raccoon-8.png    259     194  raccoon    16    11   236   175
15   raccoon-9.png    347     510  raccoon    10     7   347   471
Successfully converted xml to csv.
         filename  width  height    class  xmin  ymin  xmax  ymax
0  raccoon-16.png    424     640  raccoon    51   178   355   632
1  raccoon-17.png    259     194  raccoon    95    60   167   118
2  raccoon-18.png    240     156  raccoon    32    25   201   130
3  raccoon-19.png    259     194  raccoon    87     8   182    89
4  raccoon-20.png    720     540  raccoon     2    29   720   503
5  raccoon-21.png    290     174  raccoon    59     2   216   171
6  raccoon-22.png    640     360  raccoon   252    76   466   335
7  raccoon-23.png    259     194  raccoon   108     1   258   194
8  raccoon-24.png    268     188  raccoon    77    48   179   156
9  raccoon-24.png    268     188  raccoon   139    77   202   145
Successfully converted xml to csv.

希望这有帮助