Homec4science
Diffusion Motion correction paper (master) Browse

R6289/make_video.py
cbac9505fc6fmaster

Last Change
Raw File

make_video.py
View Options

from skimage import io
from matplotlib import pyplot as plt
from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
from multiprocessing import Pool
import numpy as np
from PIL import Image, ImageDraw, ImageFont
exp1 = io.imread('/mnt/internal_hdd/aymanns/Chen2018/columns/1_170426_fly1_013/l300g4/column.tif')
exp2 = io.imread('/mnt/internal_hdd/aymanns/Chen2018/columns/2_170608_fly1_007/l500g0/column.tif')
exp4 = io.imread('/mnt/internal_hdd/aymanns/Chen2018/columns/4_170609_fly1_000/l1000g4/column.tif')
outdir = '/mnt/internal_hdd/aymanns/Chen2018/columns/'
def plot_frame(i):
img = np.zeros((1536,1536,3))
img[:,:512,:]=exp2[i]
img[:,512:1024,:]=exp4[i]
img[:,1024:1536,:]=exp1[i]
#img[black_text_idx]=0
#img[white_text_idx]=255
image = Image.fromarray(img.astype('uint8'))
draw = ImageDraw.Draw(image)
font = ImageFont.truetype("FreeSans.ttf", size=45)
draw.text((12,12), 'Raw', fill='rgb(255,255,255)', font=font)
draw.text((12,524), 'Transformation', fill='rgb(0,0,0)', font=font)
draw.text((12,1036), 'Registered', fill='rgb(255,255,255)', font=font)
draw.text((400,12), 'Fly 1', fill='rgb(255,255,255)', font=font)
draw.text((912,12), 'Fly 2', fill='rgb(255,255,255)', font=font)
draw.text((1424,12), 'Fly 3', fill='rgb(255,255,255)', font=font)
image.save(outdir+'frame{}.png'.format(i))
#io.imsave(outdir+'frame{}.png'.format(i), img.astype('uint8'))
#plt.imshow(exp1[i],exp2[i])
#ax1 = plt.subplot2grid((1536,1536),(0,0), colspan=1536, rowspan=512)
#ax1.
#ax1.imshow(exp1[i],origin='upper')
#plt.gca().text(12,12,'Raw',fontsize=12)
#ax1.text(12,524,'Transformation',fontsize=12,color='white')
#ax1.text(12,1036,'Registered',fontsize=12,color='black')
#ax1.text(480,12,'Fly 1',fontsize=12,color='white')
#ax2 = plt.subplot2grid((1536,1536),(0,512), colspan=1536, rowspan=512)
#ax2.imshow(exp2[i])
#ax2.text(12,992,'Fly 2',fontsize=12,color='white')
#ax3 = plt.subplot2grid((1536,1536),(0,1024), colspan=1536, rowspan=512)
#ax3.imshow(exp1[i])
#ax3.text(12,1504,'Fly 3',fontsize=12,color='white')
#plt.axis('off')
#plt.savefig(outdir+'frame{}.png'.format(i), facecolor=fig.get_facecolor(), edgecolor='none', transperent=True, bbox_inches='tight')
#plt.close(fig)
#fig = plt.figure(num=None, figsize=(3, 3), dpi=512, facecolor='black')
#canvas = FigureCanvas(fig)
#ax = fig.gca()
##ax.axis('off')
##fig.tight_layout()
#ax.text(12,524,'Transformation',fontsize=12,color='white')
#ax.text(480,12,'Fly 1',fontsize=12,color='white')
#ax.text(12,992,'Fly 2',fontsize=12,color='white')
#ax.text(12,1504,'Fly 3',fontsize=12,color='white')
#canvas.draw()
#white_text = np.reshape(np.fromstring(canvas.tostring_rgb(), dtype='uint8'),(1536,1536,3))
#io.imsave('white_text.png',white_text)
#white_text_idx = np.where(white_text>0)
#fig = plt.figure(num=None, figsize=(3, 3), dpi=512, facecolor='white')
#canvas = FigureCanvas(fig)
#ax = fig.gca()
#ax.axis('off')
#fig.tight_layout()
#ax.text(12,1036,'Registered',fontsize=12,color='black')
#canvas.draw()
#black_text = np.reshape(np.fromstring(canvas.tostring_rgb(), dtype='uint8'),(1536,1536,3))
#black_text_idx = np.where(black_text < 255)
pool = Pool()
#pool.map(plot_frame,range(1))
pool.map(plot_frame,range(len(exp1)))
c4science · Help