import xml.etree.ElementTree as ET


def parse_rec(filename):
    """Parse a PASCAL VOC xml file."""
    tree = ET.parse(filename)
    objects = []
    for obj in tree.findall('object'):
        obj_struct = {}
        obj_struct['name'] = obj.find('name').text
        rbox = obj.find('rbox')
        obj_struct['rbox'] = [float(rbox.find('cx').text),
                              float(rbox.find('cx').text),
                              float(rbox.find('w').text),
                              float(rbox.find('h').text),
                              float(rbox.find('angle').text)]
        objects.append(obj_struct)
    return objects

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐