def loadSamples(directory): samples = [] for sampleFileName in (os.listdir(directory)): sampleExtension = sampleFileName[-4:] if sampleExtension != '.tif': #print("" % (fileExtension)) continue sample = Image.open(directory + '/' + sampleFileName) if not samples: samples.append(((0, 0), sample)) else: estimator = MotionEstimator(samples[0][1], sample) (x, y) = map(int, estimator.offset()) print("%s: (%d, %d)" % (sampleFileName, x, y)) samples.append(((x, y), sample)) return samples
def loadSamples(directory): samples = [] for sampleFileName in (os.listdir(directory)): print(sampleFileName) sampleExtension = sampleFileName[-4:] if sampleExtension != '.tif': #print("" % (fileExtension)) continue sample = Image.open(directory + '/' + sampleFileName) if not samples: samples.append(((0, 0), sample)) else: estimator = MotionEstimator(samples[0][1], sample) (x, y) = map(int, estimator.offset()) print ("%s: (%d, %d)" % (sampleFileName, x, y)) samples.append(((x, y), sample)) return samples
sys.exit(0) samples_folder = sys.argv[1] config = myconfig.config print "Estimate Motion Between Images" input_images = [] for file in (os.listdir(samples_folder)): if file[-4:] != '.tif': continue image = Image.open(samples_folder + '/' + file) if not input_images: input_images.append(((0, 0), image)) else: estimator = MotionEstimator(input_images[0][1], image) (x, y) = map(int, estimator.offset()) print "%s: (%d, %d)" % (file, x, y) input_images.append(((x, y), image)) scale = config['scale'] camera = Camera.Camera(config['psf']) print "Restore SR Image" # start value = sum(upsampled + shifted LR) high_res_size = [int(input_images[0][1].size[1] * scale), int(input_images[0][1].size[0] * scale), 3] high_res_image = numpy.zeros(high_res_size).astype(numpy.float32) for (offset, LR_img) in input_images: dx, dy = offset HR_arr = numpy.asarray(LR_img.resize((high_res_size[1],high_res_size[0]), Image.ANTIALIAS)) high_res_image += numpy.dstack((camera.doOffset(HR_arr[:,:,0],(-dx,-dy)),
sys.exit(0) samples_folder = sys.argv[1] config = myconfig.config print "Estimate Motion Between Images" input_images = [] for file in (os.listdir(samples_folder)): if file[-4:] != '.tif': continue image = Image.open(samples_folder + '/' + file) if not input_images: input_images.append(((0, 0), image)) else: estimator = MotionEstimator(input_images[0][1], image) (x, y) = map(int, estimator.offset()) print "%s: (%d, %d)" % (file, x, y) input_images.append(((x, y), image)) scale = config['scale'] camera = Camera.Camera(config['psf']) print "Restore SR Image" # start value = sum(upsampled + shifted LR) high_res_size = [ int(input_images[0][1].size[1] * scale), int(input_images[0][1].size[0] * scale), 3 ] high_res_image = numpy.zeros(high_res_size).astype(numpy.float32) for (offset, LR_img) in input_images: