Example #1
0
 def show_transform(self, transform_, date_='new'):
     logging.info('Showing transform: ' +
                  self._fmt_transform(transform_, date_))
     self._transform_date = date_
     self._selected_transform = transform_
     self.im_reg = warp_int(self.im1_sel.img, self._selected_transform.inverse)
     self._checkerboard()
    transforms = np.recfromcsv(fname, names=None)

    imgs = []

    for t in transforms:
        data = list(t)
        date, target, ref = data[:3]

        imgs.append((target, ref, data[3:]))


    for img, ref, params in imgs:
        im = io.imread(os.path.join(path_src,img))
        matrix = np.array(params).reshape(3,3)
        trans = transform.AffineTransform(matrix=matrix)
        img_reshaped = warp_int(im, trans.inverse)
        
        save_dir = os.path.join(path_target, ref)
        if not os.path.exists(save_dir):
            os.makedirs(save_dir)

        file_path = os.path.join(save_dir, img)
        i = 1
        
        while os.path.exists(file_path):
            root, ext = os.path.splitext(img)
            img_name = "".join([root , "_reg%d" % i, ext])

            file_path = os.path.join(save_dir, img_name)
            i+=1
Example #3
0
def transform_and_compare(img, img_ref, params, f_cmp=mean_sq_diff):
    new_transform = get_transform(params)
    img_new = warp_int(img, new_transform.inverse)
    return f_cmp(img_new, img_ref)