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

时间:2018-08-27 22:50:59

标签: python xml csv object-detection

为了使用tensorflow检测API,我需要将我的xml文件转换为csv标签。我使用了从datitran的浣熊检测模型中获取的代码。他将其放在github上,尝试一下。 https://github.com/datitran/raccoon_dataset

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

x >> y == x >>= (\_ -> y)

其中“火车”和“测试”是带有xml文件的文件夹,图像是包含它们的文件夹,数据是我希望保存标签文件的文件夹。

1 个答案:

答案 0 :(得分:1)

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

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

,如果我假设您想使用xml文件夹中的annotations文件来创建新的csv文件来分别包含测试和训练数据,那么您已经拥有了要做的是在test目录中创建两个名为trainImages的新目录,并将xml中的annotations文件复制到您创建的这些新目录中,然后运行上面的代码,即可正常运行

我只是这样做了,只复制了xml目录中的annotations个文件 而当我在打印数据框以显示

 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.

希望这会有所帮助