/
xmls_yolo.py
62 lines (60 loc) · 1.68 KB
/
xmls_yolo.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from lxml import etree
def beginXML(img, width, height, depth):
annotation = etree.Element('annotation')
fo = etree.Element('folder')
fo.text = '/test/training/images'
annotation.append(fo)
f = etree.Element('filename')
f.text = str(img)
annotation.append(f)
size = etree.Element('size')
w = etree.Element('width')
w.text = str(width)
h = etree.Element('height')
h.text = str(height)
d = etree.Element('z')
d.text = str(depth)
c = etree.Element('depth')
c.text = str(3)
size.append(w)
size.append(h)
size.append(d)
size.append(c)
annotation.append(size)
return annotation
def createXML(xmin, xmax, ymin, ymax, zmin, zmax, img, width, height, depth, annotation):
object = etree.Element('object')
n = etree.Element('name')
p = etree.Element('pose')
t = etree.Element('truncated')
d_1 = etree.Element('difficult')
bb = etree.Element('bndbox')
n.text = 'NGpair'
p.text = 'center'
t.text = str(1)
d_1.text = str(0)
xmi = etree.Element('xmin')
ymi = etree.Element('ymin')
xma = etree.Element('xmax')
yma = etree.Element('ymax')
zmi = etree.Element('zmin')
zma = etree.Element('zmax')
xmi.text = str(xmin)
yma.text = str(ymax)
ymi.text = str(ymin)
xma.text = str(xmax)
zmi.text = str(zmin)
zma.text = str(zmax)
bb.append(xmi)
bb.append(ymi)
bb.append(xma)
bb.append(yma)
bb.append(zmi)
bb.append(zma)
object.append(n)
object.append(p)
object.append(t)
object.append(d_1)
object.append(bb)
annotation.append(object)
return annotation