python xml 다루기...

Question

xml 입니다. 직접적인 이미지 정보와 바운딩 박스 정보만 가져왓습니다. 이 녀석의 image태그에 보시면 id, name, width, height가 있는데 태그를 가져오는 건알겠는데 속성 은 어떻게 가져오는지 모르겠네요예시로 width값 하나만 뽑을때 어떻게 하는지 알려주시면 감사드리겠습니다.아래는 지금 짜고 있는 코드 입니다.import os import glob from xml.etree.ElementTree import parse xml_dir = "./data/" class Voc_to_yolo_convter(): def __init__(self, xml_path): self.xml_path_list = glob.glob(os.path.join(xml_path, "*.xml")) print(self.xml_path_list) def get_voc_to_yolo(self): for xml_path in self.xml_path_list: tree = parse(xml_path) root = tree.getroot() meta = root.findall("image") test= Voc_to_yolo_convter(xml_dir) test.get_voc_to_yolo()

Clickin · Answer

xmltodict 라이브러리를 사용하시는걸 강력하게 추천드립니다.익숙한 dict 형태로 xml 사용하실 수 있습니다.

rarity · Answer

import glob from xml.etree.ElementTree import parse xml_dir &#61; &#34;./&#34; class Voc_to_yolo_convter(): def __init__(self, xml_path): self.xml_path_list &#61; glob.glob(os.path.join(xml_path, &#34;*.xml&#34;)) print(self.xml_path_list) def get_voc_to_yolo(self): for xml_path in self.xml_path_list: tree &#61; parse(xml_path) root &#61; tree.getroot() root_image_list&#61; root.findall('image') meta&#61;[] for image in root_image_list: data&#61;{} data['id']&#61;image.attrib['id'] data['width']&#61;image.attrib['width'] data['height']&#61;image.attrib['height'] meta.append(data) return meta test&#61; Voc_to_yolo_convter(xml_dir) meta&#61;test.get_voc_to_yolo() meta xml의 경우 image 태그들의 부모 태그 하나만 추가 작성하였습니다.

f10024 · Answer

import xml.etree.ElementTree as ET tree &#61; ET.parse('test.xml') # 해당 트리의 root를 반환 root &#61; tree.getroot() for child in root: # image image &#61; child.tag attr &#61; child.attrib print(image, '	' , attr) # 속성 예시 attr['id'] for sub_chid in child: box &#61; sub_chid.tag attr &#61; sub_chid.attrib print('	', box, '	' , attr) # 속성 예시 attr['label']

python xml 다루기...

3개의 답변

#yolo 목록