plot_average_image.py

import os
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image
from tqdm import tqdm


def file_list(path):
  return list(map(lambda s: os.path.join(path, s), os.listdir(path)))


def plot_average_image(path):
  a = file_list(path)
  print(len(a), a[:5])
  a_ = None
  n = 1
  for i in tqdm(a):
    try:
      img = Image.open(i, 'r').resize((228, 228))
    except:
      print('Wrong image file:', i)
      continue
    img = np.array(img)
    if n == 1:
      a_ = img
    else:
      a_ = ((a_ * n) + img) / (n + 1)
    n += 1
    del img
  plt.imshow(a_ / 255.)
  plt.axis('off')
  plt.show()
  del a_
  del a



plot_average_image('../animeface/images')
plot_average_image('../celebA/img_align_celeba/img_align_celeba')