60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
Created on Thu Apr 12 14:13:34 2018
|
||
|
|
|
||
|
|
@author: Armando
|
||
|
|
"""
|
||
|
|
import ReadIM
|
||
|
|
from skimage import exposure
|
||
|
|
import cv2
|
||
|
|
import numpy as np
|
||
|
|
from functions_utile import draw_rectangle, image_correlation, test_homography, magnification
|
||
|
|
|
||
|
|
"""Traiter juste une image aves la calibration déjà faite :D"""
|
||
|
|
working_directory = "E:\\Donnees\\ArmandoBulle\\python_treatement\\PLIF_test\\"
|
||
|
|
|
||
|
|
#If you are working with DaViS here is where you write the name of the file
|
||
|
|
image = []
|
||
|
|
vbuff, vatts = ReadIM.extra.get_Buffer_andAttributeList(working_directory + 'donnees\\20180312_testeauXG25ppm_1\\B00350.im7')
|
||
|
|
v_array, vbuff = ReadIM.extra.buffer_as_array(vbuff)
|
||
|
|
img_rescaled = exposure.rescale_intensity(v_array)
|
||
|
|
img_rescaled = img_rescaled[0]
|
||
|
|
image = img_rescaled
|
||
|
|
|
||
|
|
image_left= image.copy()
|
||
|
|
image_left, rectangle_left = draw_rectangle(image_left)
|
||
|
|
|
||
|
|
#Obtain the image zone right
|
||
|
|
image_right= image.copy()
|
||
|
|
image_right, rectangle_right = draw_rectangle(image_right)
|
||
|
|
|
||
|
|
#Obtain the homography between both images
|
||
|
|
h , status = image_correlation(image_left,image_right)
|
||
|
|
image_right = cv2.warpPerspective(image_right, h, (image_left.shape[1],image_left.shape[0]))
|
||
|
|
|
||
|
|
image_ratio = np.divide(image_right,image_left)
|
||
|
|
|
||
|
|
|
||
|
|
for y in range(len(image_ratio)-1):
|
||
|
|
for x in range(len(image_ratio[0])-1):
|
||
|
|
if np.isinf(image_ratio[y][x]):
|
||
|
|
image_ratio[y][x] = 0
|
||
|
|
elif np.isnan(image_ratio[y][x]):
|
||
|
|
image_ratio[y][x] = 0
|
||
|
|
cv2.namedWindow("Final Source Image",cv2.WINDOW_NORMAL)
|
||
|
|
cv2.imshow("Final Source Image", image_ratio)
|
||
|
|
cv2.waitKey(0)
|
||
|
|
cv2.destroyAllWindows()
|
||
|
|
|
||
|
|
image_ratio = cv2.cvtColor(image_ratio, cv2.COLOR_GRAY2BGR)
|
||
|
|
|
||
|
|
cv2.imwrite(working_directory + 'Super_position_images_test.jpg', image_ratio)
|
||
|
|
cv2.imwrite(working_directory + 'Super_position_images_test.png', image_ratio)
|
||
|
|
cv2.imwrite(working_directory + 'Super_position_images_test.tiff', image_ratio)
|
||
|
|
|
||
|
|
rectangle_final = draw_rectangle(image_ratio)[1]
|
||
|
|
|
||
|
|
image_ratio =image_ratio[rectangle_final[0][1]:rectangle_final[1][1],rectangle_final[0][0]:rectangle_final[1][0]]
|
||
|
|
|
||
|
|
cv2.imwrite(working_directory + 'Super_position_images_test2.tiff', image_ratio)
|