将xml中的中文标签转为英文

最新推荐文章于 2023-07-26 17:27:47 发布

carry_hjr

最新推荐文章于 2023-07-26 17:27:47 发布

阅读量555

点赞数

分类专栏：遥感图像处理与深度学习

本文链接：https://blog.youkuaiyun.com/carry_hjr/article/details/118661654

版权

遥感图像处理与深度学习专栏收录该内容

22 篇文章

订阅专栏

该博客使用Python的ElementTree库解析XML文件，从VOC2007数据集中提取图像对象的名称和边界框信息。博客内容涉及图像识别和数据标注，将对象名称转换为拼音并写入文本文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

别问，就是写着玩

from pypinyin import lazy_pinyin

for image_id in image_ids:
    with open("./input/ground-truth/"+image_id+".txt", "a", encoding='utf8') as new_f:
        root = ET.parse("/home/yons/data/huawei/shenzhen/trainval/VOC2007/Annotations/"+image_id+".xml").getroot()
        for obj in root.findall('object'):
            obj_name = obj.find('name').text
            obj_name = '_'.join(lazy_pinyin(obj_name))
            bndbox = obj.find('bndbox')
            left = bndbox.find('xmin').text
            top = bndbox.find('ymin').text
            right = bndbox.find('xmax').text
            bottom = bndbox.find('ymax').text
            new_f.write("%s %s %s %s %s\n" % (obj_name, left, top, right, bottom))
print("Conversion completed!")