whatnick

#!python

"""
This script is designed to register RGB and IR
images acquired from different cameras with different
orientation and produce a 4-band subset.
"""

import pylab as plt
import numpy as np
import PIL.Image as im
import sys

def register_images(ir_image,rgb_image):
    """
    Implement some sort of global registration
    """
    t_xsize = 600
    t_ysize = 385

    f_xsize = 640
    f_ysize = 480
    return t_xsize,t_ysize,f_xsize,f_ysize

if __name__=="__main__":

    if(len(sys.argv)==3):
        ir_image = plt.imread(sys.argv[1])
        rgb_image = plt.imread(sys.argv[2])
    else:
        ir_image = plt.imread("0.jpg")
        rgb_image = plt.imread("1.jpg")

    t_xsize,t_ysize,f_xsize,f_ysize = register_images(ir_image, rgb_image)

    cir_image = np.zeros((t_ysize,t_xsize,3),dtype="int8")

    ir_subset = ir_image[f_ysize-t_ysize:f_xsize,0:t_xsize,:]
    rgb_subset = rgb_image[0:t_ysize,f_xsize-t_xsize:f_xsize,:]

    cir_image[:,:,0] = ir_subset[:,:,0]
    cir_image[:,:,1] = rgb_subset[:,:,0]
    cir_image[:,:,2] = rgb_subset[:,:,1]

    ir_im = im.fromarray(ir_subset)
    rgb_im = im.fromarray(rgb_subset)
    cir_im = im.fromarray(cir_image)

    cir_im = cir_im.transpose(im.FLIP_TOP_BOTTOM)
    ir_im = ir_im.transpose(im.FLIP_TOP_BOTTOM)
    rgb_im = rgb_im.transpose(im.FLIP_TOP_BOTTOM)

    ir_im.save("ir_image.png")
    rgb_im.save("rgb_image.png")
    cir_im.save("cir_image.png")