| | |
|---|
| | | |
|---|
| | | |
|---|
| | | 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 |
|---|
| | |
|---|
| | | 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') |
|---|
| | |
|---|
| | | 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") |
|---|
| | | |
|---|
