Add tqdm during mean and std computation

dan/ocr/manager/dataset.py

@@ -7,6 +7,7 @@ import json
 
 import numpy as np
 from torch.utils.data import Dataset
+from tqdm import tqdm
 
 from dan.datasets.utils import natural_sort
 from dan.utils import read_image, token_to_ind
@@ -132,7 +133,7 @@ class OCRDataset(Dataset):
         diff = np.zeros((3,))
         nb_pixels = 0
         for metric in ["mean", "std"]:
-            for ind in range(len(self.samples)):
+            for ind in tqdm(range(len(self.samples)), desc=f"Computing {metric} value"):
                 img = np.array(self.get_sample_img(ind))
                 if metric == "mean":
                     total += np.sum(img, axis=(0, 1))