|
|
|
@ -10,10 +10,10 @@ classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def convert(size, box): |
|
|
|
|
dw = 1./size[0] |
|
|
|
|
dh = 1./size[1] |
|
|
|
|
x = (box[0] + box[1])/2.0 |
|
|
|
|
y = (box[2] + box[3])/2.0 |
|
|
|
|
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 |
|
|
|
@ -34,7 +34,7 @@ def convert_annotation(year, image_id): |
|
|
|
|
for obj in root.iter('object'): |
|
|
|
|
difficult = obj.find('difficult').text |
|
|
|
|
cls = obj.find('name').text |
|
|
|
|
if cls not in classes or int(difficult) == 1: |
|
|
|
|
if cls not in classes or int(difficult)==1: |
|
|
|
|
continue |
|
|
|
|
cls_id = classes.index(cls) |
|
|
|
|
xmlbox = obj.find('bndbox') |
|
|
|
@ -54,3 +54,6 @@ for year, image_set in sets: |
|
|
|
|
convert_annotation(year, image_id) |
|
|
|
|
list_file.close() |
|
|
|
|
|
|
|
|
|
os.system("cat 2007_train.txt 2007_val.txt 2012_train.txt 2012_val.txt > train.txt") |
|
|
|
|
os.system("cat 2007_train.txt 2007_val.txt 2007_test.txt 2012_train.txt 2012_val.txt > train.all.txt") |
|
|
|
|
|
|
|
|
|