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

Question

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

218.xml

`<annotation>
    <folder>train</folder>
    <filename>218</filename>
    <path>C:\kp\images\train\218.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 的输入。我已经浏览过Caffe LMDB train and val.txt , how to make train.txt file in caffe , A guide to convert_imageset.cpp , https://github.com/chuanqi305/MobileNet-SSD/issues/90

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

Answer 1

火车.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