def dump_images(prefix):
# run from imgs -> bitmap and stitch them together...
img_collage = Image.new('RGB', (17 * 8, 17 * 8), (0, 0, 0))
bitmap_collage = Image.new('RGB', (9 * 8, 9 * 8), (255, 255, 255))
centroids_collage = Image.new('RGB', (9 * 8, 9 * 8), (255, 255, 255))
ims, bs = sess.run([imgs, model.output])
for x in range(8):
for y in range(8):
i = (x * 8) + y
img_collage.paste(u.zero_centered_array_to_pil_image(ims[i]),
(17 * x, 17 * y))
output_bitmap = u.bitmap_to_pil_image(bs[i])
bitmap_collage.paste(output_bitmap, (9 * x, 9 * y))
centroids = u.centroids_of_connected_components(bs[i])
centroid_bitmap = u.bitmap_from_centroids(centroids, h=8, w=8)
centroid_bitmap = u.bitmap_to_single_channel_pil_image(
centroid_bitmap)
centroids_collage.paste(centroid_bitmap, (9 * x, 9 * y))
img_collage.save("images/ra/%s_imgs.png" % prefix)
bitmap_collage.save("images/ra/%s_bitmaps.png" % prefix)
centroids_collage.save("images/ra/%s_centroids.png" % prefix)
import os
import sys
import util as u
# TODO: make this multiprocess, too slow as is...
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--label-db',
type=str,
help='label_db to materialise bitmaps from')
parser.add_argument('--directory', type=str, help='directory to store bitmaps')
opts = parser.parse_args()
print(opts)
label_db = LabelDB(label_db_file=opts.label_db)
if not os.path.exists(opts.directory):
os.makedirs(opts.directory)
fnames = list(label_db.imgs())
for i, fname in enumerate(fnames):
bitmap = u.xys_to_bitmap(xys=label_db.get_labels(fname),
height=1024,
width=768,
rescale=0.5)
single_channel_img = u.bitmap_to_single_channel_pil_image(bitmap)
single_channel_img.save("%s/%s" %
(opts.directory, fname.replace(".jpg", ".png")))
sys.stdout.write("%d/%d \r" % (i, len(fnames)))