/Python/Convert pascal annotation to yolo

Convert pascal annotation to yolo

python convert.py PATH_TO_ANNOTATION

# convert.py 
import xml.etree.ElementTree as ET
import pickle
import os,sys
from os import listdir, getcwd
from os.path import join

classes=["08901725101053","08901030664205","08901030662010","04902430772877","08901030612381","08901512540102"]


def convert(size,box):
	# Conversion from VOC to YOLO format
	dw = 1./(size[0])
	dh = 1./(size[1])
	x = (box[0] + box[1])/2.0 - 1
	y = (box[2] + box[3])/2.0 - 1
	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)

#print ann_file
def converting_annotation(ann_file):
	for ann in ann_file:
		txt_file= ann.split('.')[0]+'.txt'
		in_file=open(ann_dir+ann)
		out_file =open(ann_dir+txt_file, '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)

		print w,h
		for obj in root.iter('object'):
			cls=obj.find('name').text
			if cls not in classes:
				continue
			cls_id=classes.index(cls)
			xmlbox = obj.find('bndbox')
			b = (float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text), float(xmlbox.find('ymax').text))
			data = convert((w,h), b)
			out_file.write(str(cls_id) + " " + " ".join([str(a) for a in data]) + '\n')


if __name__ == '__main__': 
	ann_dir=sys.argv[1]
	ann_file=os.listdir(ann_dir)
	ann_file.sort()
	print(ann_file)
	converting_annotation(ann_file)

Source: github.com

Similar codes

3 votes339 views1 answer
Auto py to exe\
Python
0 vote357 views1 answer
Y=mx+b python
Python
24 votes365 views1 answer
Creating a new enviroment in conda
Python
0 vote337 views1 answer
Number of unique pairs in columns pandas
Python
0 vote356 views1 answer
Python how to make integer show 2 numbers
Python
0 vote352 views2 answers
SyntaxError: Non-ASCII character '\xe2' in file /root/Downloads/la_southwest_college/handy/accountDetails.py on line 116, but no encoding declared; see http://python.org/dev/peps/pep-0263/ for details
Python
3 votes329 views1 answer
Update windows wallpaper python
Python
0 vote373 views1 answer
How to convert a list to a string by newline python
Python
0 vote363 views1 answer
Mean =[0,0] covariance = [[1,0],[0,100]] ds = np.random.multivariate_normal(mean,covariance,500) dframe = pd.DataFrame(ds, columns=['col1', 'col2']) fig = sns.kdeplot(dframe).get_figure() fig.savefig('kde1.png')
Python
2 votes361 views1 answer
How to move the pointer on screen using python
Python