Commit 4e0db0cda2d9597119f2bb15ab5538130bf4e3be
1 parent
6e774c03
added imagestack source code
Showing
1 changed file
with
40 additions
and
0 deletions
Show diff stats
1 | +import glob | ||
2 | +import imageio | ||
3 | +import numpy | ||
4 | +import progressbar | ||
5 | + | ||
6 | +def loadstack(fmask, dtype=numpy.float32): | ||
7 | + #read the input files | ||
8 | + F = glob.glob(fmask) | ||
9 | + if(len(F)==0): | ||
10 | + print("No image files found.") | ||
11 | + return | ||
12 | + | ||
13 | + #load the first file | ||
14 | + I = imageio.imread(F[0]) | ||
15 | + | ||
16 | + #generate the image stack | ||
17 | + S = numpy.zeros((len(F), I.shape[0], I.shape[1], I.shape[2]), dtype) | ||
18 | + | ||
19 | + bar = progressbar.ProgressBar(max_value=len(F)) | ||
20 | + | ||
21 | + for i in range(0, len(F)): | ||
22 | + I = imageio.imread(F[i]) | ||
23 | + S[i, :, :, :] = I | ||
24 | + bar.update(i+1) | ||
25 | + | ||
26 | + return S | ||
27 | + | ||
28 | +def rgb2gray(rgb): | ||
29 | + r, g, b = rgb[...,0], rgb[...,1], rgb[...,2] | ||
30 | + gray = 0.2989 * r + 0.5870 * g + 0.1140 * b | ||
31 | + | ||
32 | + return gray | ||
33 | + | ||
34 | +def savestack(I, fname, extension = ".bmp"): | ||
35 | + #output the final images | ||
36 | + bar = progressbar.ProgressBar(max_value=I.shape[0]) | ||
37 | + #save the output files | ||
38 | + for i in range(0, I.shape[0]): | ||
39 | + imageio.imwrite(fname + "%03d" % i + extension, I[i, :, :, :]) | ||
40 | + bar.update(i+1) | ||
0 | \ No newline at end of file | 41 | \ No newline at end of file |