def save_compressed_img(k, closest_points, centroids, rows, cols, colors, img, output=None, iteration=''):
img_arr = np.zeros((rows*cols, colors))
for i in range(k):
img_arr[np.where(closest_points == i),:] = centroids[i]
compressed_img = np.reshape(img_arr, (rows, cols, colors))
if not output:
img_ls = img.split('.')
misc.imsave(str(k)+'k_compressed_' + ''.join(img_ls[:-1]) + iteration + '.' + img_ls[-1], compressed_img)
return os.stat(str(k)+'k_compressed_' + ''.join(img_ls[:-1]) + iteration+ '.' + img_ls[-1]).st_size
else:
output_ls = output.split('.')
misc.imsave(''.join(output_ls[:-1]) + iteration + '.' + output_ls[-1], compressed_img)
return os.stat(''.join(output_ls[:-1]) + iteration + '.' + output_ls[-1]).st_size
def load_and_process(path,
locs,
ids,
prefix="_home_moffatopera_",
suffix='_binary.png',
plot=False):
"""asdfsadgsahfd"""
if hasattr(locs, 'files'):
iterable_locs = locs.files
else:
iterable_locs = locs
#import progressbar as pbar
#widg = widgets=[pbar.RotatingMarker(), ' ', pbar.Percentage(),' ', \
# pbar.Bar(), ' ', pbar.ETA()]
#pb = pbar.ProgressBar(maxval=len(iterable_locs), widgets=widg).start()
#count = 0
allobjects = []
allids = []
for image in iterable_locs:
print type(ids)
imroot = image.split('.')[0]
im = imread_binary(os.path.join(path, prefix + imroot + suffix))
im_locs = locs[image]
im_ids = ids[image]
print prefix + imroot + suffix
if plot:
aligned_objects_from_im(im, im_locs, im_ids, image, plot=plot)
print "Hit enter for next image."
else:
objects, newids = aligned_objects_from_im(im,
im_locs,
im_ids,
image,
plot=plot)
#count += 1
if not plot:
allobjects.append(objects)
allids.append(newids)
#pb.update(count)
if not plot:
allobjects = np.concatenate(allobjects, axis=0)
allids = np.concatenate(allids, axis=0)
return allobjects, allids
def load_and_process(path, locs, ids, prefix="_home_moffatopera_",
suffix='_binary.png', plot=False):
"""asdfsadgsahfd"""
if hasattr(locs,'files'):
iterable_locs = locs.files
else:
iterable_locs = locs
#import progressbar as pbar
#widg = widgets=[pbar.RotatingMarker(), ' ', pbar.Percentage(),' ', \
# pbar.Bar(), ' ', pbar.ETA()]
#pb = pbar.ProgressBar(maxval=len(iterable_locs), widgets=widg).start()
#count = 0
allobjects = []
allids = []
for image in iterable_locs:
print type(ids)
imroot = image.split('.')[0]
im = imread_binary(os.path.join(path,prefix+imroot+suffix))
im_locs = locs[image]
im_ids = ids[image]
print prefix+imroot+suffix
if plot:
aligned_objects_from_im(im,im_locs,im_ids,image,plot=plot)
print "Hit enter for next image."
else:
objects, newids = aligned_objects_from_im(im,im_locs,im_ids,image,
plot=plot)
#count += 1
if not plot:
allobjects.append(objects)
allids.append(newids)
#pb.update(count)
if not plot:
allobjects = np.concatenate(allobjects,axis=0)
allids = np.concatenate(allids,axis=0)
return allobjects, allids
ppgplot.pgsci(1)
ppgplot.pgenv(0, width-1, 0, height-1, 0, 0)
outputWindow = ppgplot.pgopen('/xs')
ppgplot.pgask(False)
pgPlotTransform = [0, 1, 0, 0, 0, 1]
ppgplot.pgsci(1)
ppgplot.pgenv(0, width-1, 0, height-1, 0, 0)
redReference = numpy.array(images[0].split()[0])
redTotal = numpy.zeros(numpy.shape(redReference))
greenTotal = numpy.zeros(numpy.shape(redReference))
blueTotal = numpy.zeros(numpy.shape(redReference))
for frameIndex, image in enumerate(images):
redData, greenData, blueData = image.split()
# redData.show()
# print redData
redData = numpy.array(redData)
ppgplot.pgslct(previewWindow)
ppgplot.pggray(redData, 0, width-1, 0, height-1, 0, 255, pgPlotTransform)
# correlation = scipy.signal.correlate2d(redData, redReference, mode='same', boundary='symm')
shift = phase_cor(redData, redReference)
y, x = numpy.unravel_index(numpy.argmax(shift), shift.shape)
if x > width/2:
xd = width - x
else:
xd = -x
(width, height) = image.size
previewWindow = ppgplot.pgopen('/xs')
ppgplot.pgask(False)
pgPlotTransform = [0, 1, 0, 0, 0, 1]
ppgplot.pgsci(1)
ppgplot.pgenv(0, width-1, 0, height-1, 0, 0)
outputWindow = ppgplot.pgopen('/xs')
ppgplot.pgask(False)
pgPlotTransform = [0, 1, 0, 0, 0, 1]
ppgplot.pgsci(1)
ppgplot.pgenv(0, width-1, 0, height-1, 0, 0)
redReference, greenReference, blueReference = image.split()
redTotal = numpy.zeros(numpy.shape(redReference))
greenTotal = numpy.zeros(numpy.shape(redReference))
blueTotal = numpy.zeros(numpy.shape(redReference))
redTotal+= numpy.array(redReference)
greenTotal+= numpy.array(greenReference)
blueTotal+= numpy.array(blueReference)
for frameIndex in range(1, numImages):
f = filesToLoad[frameIndex]
image = Image.open(f)
exposureTime = getExifValue(image, 'ExposureTime')
print frameIndex,"/", numImages-1, " ", f, image.bits, image.size, image.format, image.mode, "Exposure time:", exposureTime
redData, greenData, blueData = image.split()
redData = numpy.array(redData)
