preprocess_vot.py 文件源码

def process_vot(path, min_height, min_width):
    images = []
    for dirpath, dirnames, filenames in os.walk(path):
        img_shape = None
        pad_height = 0
        pad_width = 0
        for filename in filenames:
            if filename[-4:] == ".jpg" and "_ds" not in filename:
                full_path = os.path.join(dirpath, filename)
                img = misc.imread(full_path,mode='RGB')
                img_shape = img.shape
                ratio = min(float(min_width)/img.shape[1], float(min_height)/img.shape[0])
                img = misc.imresize(img, size=ratio)
                img, pad_height, pad_width = pad_image(img, (min_height, min_width))
                output_filename = os.path.join(dirpath, filename[:-4] + "_ds.jpg")
                misc.imsave(output_filename, img)
                images.append(output_filename)
        if img_shape:
            gt_path = os.path.join(dirpath, "groundtruth.txt")
            preprocess_label(gt_path, ratio, img_shape, min_height, min_width, pad_height, pad_width)
    return images