如何将 annotation.xml 文件转换为 train.txt 和 val.txt 以进行对象检测？

Question

我正在使用 caffe 训练用于对象检测的自定义图像数据集。在这里我有火车和验证目录。通过使用 LabelImg ，对象注释在 .xml 文件中分别为每个图像创建（image1.xml、image2.xml 等）。现在我想将所有这些 xml 文件转换为 train.txt 和 val.txt。我该怎么做？我坚持这一步已经 3 周了

218.xml

`<annotation>
    <folder>train</folder>
    <filename>218</filename>
    <path>C:\kp\images\train8.jpg</path>
    <source>
        <database>Unknown</database>
    </source>
    <size>
        <width>352</width>
        <height>288</height>
        <depth>3</depth>
    </size>
    <segmented>0</segmented>
    <object>
        <name>person</name>
        <pose>Unspecified</pose>
        <truncated>0</truncated>
        <difficult>0</difficult>
        <bndbox>
            <xmin>79</xmin>
            <ymin>56</ymin>
            <xmax>262</xmax>
            <ymax>283</ymax>
        </bndbox>
    </object>
</annotation>`

我使用该工具为每个图像创建了 annotation.xml 文件 (image_name.xml)。现在我需要将其转换为 train.txt 和 val.txt 以输入到创建 LMDB。我已经经历了 , , , https://github.com/chuanqi305/MobileNet-SSD/issues/90

但是一切都是为了对象分类，其他人也没有解释..任何建议或帮助将不胜感激

Answer 1

train.txt

图片路径 xmlpath

image/image1.jpg label/image1.xml
image/image2.jpg label/image2.xml
image/image3.jpg label/image3.xml
image/image4.jpg label/image4.xml
...

这是 train.txt 或 test.txt

如何将 annotation.xml 文件转换为 train.txt 和 val.txt 以进行对象检测？

how to convert the annotation.xml files into train.txt and val.txt for object detection?

python

caffe

lmdb