Calculate average image in python

average_image.py

from __future__ import print_function
import os
from scipy.misc import imread, imsave
import numpy as np

if __name__=="__main__":

	total = None

	fiLi = os.listdir(".")
	for fi in fiLi:
		fiSp = os.path.splitext(fi)
		if fiSp[-1] != ".png": continue

		img = np.array(imread(fi), dtype=np.float32)
		if total is not None:
			total += img
		else:
			total = img.copy()

	total /= len(fiLi)
	img = np.array(total, dtype=np.uint8)
	print (total.shape)
	imsave("out.jpg", img)