data_manager.py 文件源码

def compute_mean_img(img_fns, img_sz):
  mean = None
  count = len(img_fns)
  for i in range(len(img_fns)):
    cv_img = cv2.imread(img_fns[i])
    if cv_img is None:
      raise ValueError(img_fns[i] + ' image read error')
    new_img = cv2.resize(cv_img, img_sz)
    if mean is None:
      mean = new_img.astype(np.float32)
    else:
      mean += new_img
  mean = mean / count
  return mean.astype(np.uint8)