Cacher is the code snippet organizer for pro developers

import cv2
import os
import numpy as np

image_dirs = [
'/share/datasets/AIC20_track2/AIC20_ReID_Simulation/image_train',
'/share/datasets/AIC20_track2/AIC20_ReID_Simulation/image_train',
'/share/datasets/AIC20_track2/AIC20_ReID_Simulation/image_test',
'/share/datasets/AIC20_track2/AIC20_ReID_Simulation/image_query'
]

rgb_arr = np.zeros(3)
no_images = 0
for image_dir in image_dirs:
  print(image_dir)
  for filename in os.listdir(image_dir):
    img = cv2.imread(os.path.join(image_dir,filename))
    rgb_arr += img.mean(axis=0).mean(axis=0)
    no_images += 1

mean = rgb_arr/no_images
print("mean rgb: ", mean)