目标检测xml文件更改大小

有时候jpg太大会导致检测的时候xunl训练过慢，需要缩小图片

注意：bag.list中./data/name.jpg应该是data/name.jpg

有两个文件夹data和data_resize

import cv2
import xml.etree.ElementTree as ET
import pdb
fid = open('bag.list')
lines = fid.readlines()
fid.close()
for line in lines:
    jpg_file = line.strip()
    jpg_id = jpg_file.split('.')[0]
    xml_file = jpg_id + '.xml'
    img = cv2.imread(jpg_file, -1)
    height_origin, width_origin = img.shape[:2]
    width = 240#int(width*0.1)
    height = 320#int(height*0.1)
    scale_x = width*1.0 / width_origin
    scale_y = height*1.0 / height_origin
    size = (width, height) 
    shrink = cv2.resize(img, size, interpolation=cv2.INTER_AREA)
    new_jpg_file = 'data_resize/' + jpg_file.split('/')[1]
    #pdb.set_trace()
    cv2.imwrite(new_jpg_file,shrink)
    
    updateTree = ET.parse(xml_file)
    root = updateTree.getroot()
    root.find("size").find('width').text = str(width)
    root.find("size").find('height').text = str(height)
    element_object = root.findall('object')
    for i in range(len(element_object)):
        xmin = element_object[i].find('bndbox').find('xmin').text
        xmax = element_object[i].find('bndbox').find('xmax').text
        ymin = element_object[i].find('bndbox').find('ymin').text
        ymax = element_object[i].find('bndbox').find('ymax').text
        element_object[i].find('bndbox').find('xmin').text = str(int(int(xmin)*scale_x))
        element_object[i].find('bndbox').find('xmax').text = str(int(int(xmax)*scale_x))
        element_object[i].find('bndbox').find('ymin').text = str(int(int(ymin)*scale_y))
        element_object[i].find('bndbox').find('ymax').text = str(int(int(ymax)*scale_y))
    new_xml_file = 'data_resize/' + xml_file.split('/')[1]
    updateTree.write(new_xml_file)
    print 'done'