Odtnhj

I have annotations in xml files such as this one, which follows the PASCAL VOC convention:

<annotation>
<folder>training</folder>
<filename>chanel1.jpg</filename>
<source>
<database>synthetic initialization</database>
<annotation>PASCAL VOC2007</annotation>
<image>synthetic</image>
<flickrid>none</flickrid>
</source>
<owner>
<flickrid>none</flickrid>
<name>none</name>
</owner>
<size>
<width>640</width>
<height>427</height>
<depth>3</depth>
</size>
<segmented>0</segmented>
<object>
<name>chanel</name>
<pose>Unspecified</pose>
<truncated>0</truncated>
<difficult>0</difficult>
<bndbox>
<xmin>344</xmin>
<ymin>10</ymin>
<xmax>422</xmax>
<ymax>83</ymax>
</bndbox>
</object>
<object>
<name>chanel</name>
<pose>Unspecified</pose>
<truncated>0</truncated>
<difficult>0</difficult>
<bndbox>
<xmin>355</xmin>
<ymin>165</ymin>
<xmax>443</xmax>
<ymax>206</ymax>
</bndbox>
</object>
</annotation>

What is the cleanest way of retrieving for example the fields filename and bndbox in Python?

I was trying to ElementTree, which seems to be the official Python solution, but I can't make it work.

My code so far:

from xml.etree import ElementTree as ET
tree = ET.parse("data/all/annotations/" + file)
fn = tree.find('filename').text
boxes = tree.findall('bndbox')

this produces

fn == 'chanel1.jpg'
boxes == 

So it succesfully extracts the filename field, but not the bndbox'es.

That's a quite easy solution for your problem:

This will return your box coordinates in a nested list [xmin, ymin, xmax, ymax] and the filename
Once I struggled with bndbox tags which where mixed up (ymin, xmin,...) or any other strange combinations, so this code read the tags not only the position.

Hope it helps...

import xml.etree.ElementTree as ET


def read_content(xml_file: str):

 tree = ET.parse(xml_file)
 root = tree.getroot()

 list_with_all_boxes = 

 for boxes in root.iter('object'):

 filename = root.find('filename').text

 ymin, xmin, ymax, xmax = None, None, None, None

 for box in boxes.findall("bndbox"):
 ymin = int(box.find("ymin").text)
 xmin = int(box.find("xmin").text)
 ymax = int(box.find("ymax").text)
 xmax = int(box.find("xmax").text)

 list_with_single_boxes = [xmin, ymin, xmax, ymax]
 list_with_all_boxes.append(list_with_single_boxes)

 return filename, list_with_all_boxes

name, boxes = read_content("file.xml")

Here is detailed documentation and source code: https://github.com/trinath503/Python/tree/master/Generate_Pascal_VOC_Files

NOTE: You need to pass the data in the specified formart

 pascal_voc_data = '''
 [
 
 "folder":"folder",
 "filename": "1.jpg",
 "path":"path",
 "source":"database":"database",
 "size":"width":256,"height":256,"depth":3,
 "segmented":0,
 "objects":["name":"name","pose":"pose","truncated":"truncated","occluded":"occluded","bndbox":"xmin":3,"xmax":33,"ymin":3,"ymax":33,"name":"name","pose":"pose","truncated":"truncated","occluded":"occluded","bndbox":"xmin":3,"xmax":33,"ymin":3,"ymax":33]
 ,
 
 "folder":"folder",
 "filename": "2.jpg",
 "path":"path",
 "source":"database":"database",
 "size":"width":256,"height":256,"depth":3,
 "segmented":0,
 "objects":["name":"name","pose":"pose","truncated":"truncated","occluded":"occluded","bndbox":"xmin":3,"xmax":33,"ymin":3,"ymax":33,"name":"name","pose":"pose","truncated":"truncated","occluded":"occluded","bndbox":"xmin":3,"xmax":33,"ymin":3,"ymax":33]
 
 ]
 '''