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:
