-
Notifications
You must be signed in to change notification settings - Fork 8
/
subtractMeanImage.py
54 lines (44 loc) · 1.49 KB
/
subtractMeanImage.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
'''
First gets the mean image of a folder. Then subtracts that mean image from every
image file and puts the new copy in a new folder.
Lars Roemheld, [email protected]
'''
__author__ = 'Lars Roemheld'
import sys, os
from scipy import misc
import numpy as np
folder = sys.argv[1]
if not os.path.exists(folder + 'minusmean/'):
os.mkdir(folder + 'minusmean/')
print "getting average"
averageIm = np.zeros((256, 256, 4))
countIm = 0
for file in os.listdir(folder):
if file.endswith(".png"):
oldFilename = folder + file
if countIm % 250 == 0:
print "now on file {}, {}".format(countIm, file)
img = misc.imread(oldFilename)
# if countIm % 250 == 0:
# new = img - averageIm
# misc.imsave('ex{}.png'.format(countIm), new)
averageIm = (countIm * averageIm + img) / (countIm + 1.0)
countIm += 1
# if countIm % 1000 == 0:
# misc.imsave('avg{}.png'.format(countIm), averageIm)
# print "saved"
misc.imsave(folder + 'folder_average.png', averageIm)
print "saved average"
print "Subtracting average from images"
countIm = 0
for file in os.listdir(folder):
if file.endswith(".png"):
oldFilename = folder + file
newFilename = folder + "minusmean/" + file
if countIm % 250 == 0:
print "now on file {}, {}".format(countIm, file)
img = misc.imread(oldFilename)
new = img - averageIm
misc.imsave(newFilename, new)
countIm += 1
print "done"