m0_64415849 2023-04-19 17:06 采纳率: 66.7%
浏览 99

将文件夹下的xml文件转换为txt到另一个文件,但是生成的txt文件是空的是为什么呢

将文件夹下的xml文件转换为txt到另一个文件,但是生成的txt文件是空的是为什么呢
xml文件内容是这样的

<annotation>
    <folder>20160523</folder>
    <filename>XRay20160523153057.bmp</filename>
    <path>\\192.168.31.118\share\ImageLabel\20160523\XRay20160523153057.bmp</path>
    <source>
        <database>Unknown</database>
    </source>
    <size>
        <width>1024</width>
        <height>768</height>
        <depth>3</depth>
    </size>
    <segmented>0</segmented>
    <object>
        <name>mobilephone</name>
        <pose>Unspecified</pose>
        <truncated>0</truncated>
        <difficult>0</difficult>
        <bndbox>
            <xmin>268</xmin>
            <ymin>384</ymin>
            <xmax>340</xmax>
            <ymax>434</ymax>
        </bndbox>
    </object>
    <object>
        <name>sword_middle</name>
        <pose>Unspecified</pose>
        <truncated>0</truncated>
        <difficult>0</difficult>
        <bndbox>
            <xmin>291</xmin>
            <ymin>405</ymin>
            <xmax>393</xmax>
            <ymax>472</ymax>
        </bndbox>
    </object>
    <object>
        <name>gun_mause</name>
        <pose>Unspecified</pose>
        <truncated>0</truncated>
        <difficult>0</difficult>
        <bndbox>
            <xmin>332</xmin>
            <ymin>352</ymin>
            <xmax>442</xmax>
            <ymax>414</ymax>
        </bndbox>
    </object>
</annotation>



import xml.etree.ElementTree as ET
import os
# 类别
CLASSES=["glassbottle","knife","middleknife","bigknife","gun","computer","mobile","bottle"]
# xml文件路径
#xml_input="D:\\temp\\yolov7-main\\data\\dataset_one\\labels\\train\\"
xml_input = "D:\\temp\\yolov7-main\\data\\dataset_one\\labels\\valid\\"
#txt_output="D:\\temp\\yolov7-main\\data\\dataset_one\\labels\\trainTXT\\"
txt_output = "D:\\temp\\yolov7-main\\data\\dataset_one\\labels\\validTXT"
def convert(size,box):
    # 将bbox的左上角点,右下角点坐标的格式,转换为bbox中心点+bbox的W,H的格式,并进行归一化
    dw=1./size[0]
    dh=1./size[1]
    x=(box[0]+box[1])/2.0
    y=(box[2]+box[3])/2.0
    w=box[1]-box[0]
    h=box[3]-box[2]
    x=x*dw
    w=w*dw
    y=y*dh
    h=h*dh
    return (x,y,w,h)
def convert_annotation(image_id):
    # 把图像image_id的xml文件转换为目标检测的label文件(txt)
    # 其中包含物体的类别cls,bbox的中心点坐标,以及bbox的W,H
    # 并将四个物理量归一化

    in_file=open(xml_input+image_id)
    image_id=image_id.split(".")[0]
    #  out_file=open(txt_output+"txt/%s.txt"%(image_id),"w")

    out_file = open(txt_output + "%s.txt" % (image_id), "w")
    tree=ET.parse(in_file)
    root=tree.getroot()
    size=root.find("size")
    w=int(size.find("width").text)
    h=int(size.find("height").text)
    for obj in root.iter("object"):
        difficult=obj.find("difficult").text
        obj_cls=obj.find("name").text
        if obj_cls not in CLASSES or int(difficult)==1:
            continue
        cls_id=CLASSES.index(obj_cls)
        xmlbox=obj.find("bndbox")
        points=(float(xmlbox.find("xmin").text),
                float(xmlbox.find("xmax").text),
                float(xmlbox.find("ymin").text),
                float(xmlbox.find("ymax").text))
        bb=convert((w,h),points)
        out_file.write(str(cls_id)+" "+" ".join([str(a) for a in bb])+"\n")
def make_label_txt():
    # labels文件夹下创建image_id.txt
    # 对应每个image_id.xml提取出的bbox信息
    filenames=os.listdir(xml_input)
    for file in filenames:
        convert_annotation(file)

if __name__=="__main__":
    # 开始提取和转换
    make_label_txt()
  • 写回答

2条回答 默认 最新

  • 守时间的孤岛 2023-04-19 17:09
    关注

    你把input内容打印一下,会不会是取出来就是空内容

    评论

报告相同问题?

问题事件

  • 创建了问题 4月19日

悬赏问题

  • ¥20 有偿:在ubuntu上安装arduino以及其常用库文件。
  • ¥15 请问用arcgis处理一些数据和图形,通常里面有一个根据点划泰森多边形的命令,直接划的弊端是只能执行一个完整的边界,但是我们有时候会用到需要在有很多边界内利用点来执行划泰森多边形的命令
  • ¥30 在wave2foam中执行setWaveField时遇到了如下的浮点异常问题,请问该如何解决呢?
  • ¥20 看图片)删除这个自动化录屏脚本就一直报错找不到脚本文件,如何解决?(相关搜索:bat文件)
  • ¥750 关于一道数论方面的问题,求解答!(关键词-数学方法)
  • ¥200 csgo2的viewmatrix值是否还有别的获取方式
  • ¥15 Stable Diffusion,用Ebsynth utility在视频选帧图重绘,第一步报错,蒙版和帧图没法生成,怎么处理啊
  • ¥15 请把下列每一行代码完整地读懂并注释出来
  • ¥15 pycharm运行main文件,显示没有conda环境
  • ¥15 寻找公式识别开发,自动识别整页文档、图像公式的软件