diff --git a/tools/eval/datasets/icdar.py b/tools/eval/datasets/icdar.py index bc2c61e5da2956123f7f1b494eaf08c6ae12b4be..524d1db725707b6e24a1ecbf2d5087a1e547c743 100644 --- a/tools/eval/datasets/icdar.py +++ b/tools/eval/datasets/icdar.py @@ -33,8 +33,9 @@ class ICDAR: def eval(self, model): right_num = 0 pbar = tqdm(self.val_label) + pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name)) + for fn, label in pbar: - pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name)) img = cv.imread(fn) diff --git a/tools/eval/datasets/iiit5k.py b/tools/eval/datasets/iiit5k.py index 0762137677ed862d7ee5566426034549a89e52d8..82b0804746f31d09f906bcda892441eaa47488a6 100644 --- a/tools/eval/datasets/iiit5k.py +++ b/tools/eval/datasets/iiit5k.py @@ -36,8 +36,9 @@ class IIIT5K: def eval(self, model): right_num = 0 pbar = tqdm(self.val_label) + pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name)) + for img, value in pbar: - pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name)) rbbox = np.array([0, img.shape[0], 0, 0, img.shape[1], 0, img.shape[1], img.shape[0]]) diff --git a/tools/eval/datasets/imagenet.py b/tools/eval/datasets/imagenet.py index 22c33101aa2c1e9d29b04e179eb7f5da6e3fdf62..571a89e9fb65fbff72cc696ae3785d18cf229dc4 100644 --- a/tools/eval/datasets/imagenet.py +++ b/tools/eval/datasets/imagenet.py @@ -39,8 +39,9 @@ class ImageNet: top_1_hits = 0 top_5_hits = 0 pbar = tqdm(self.val_label) + pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name)) + for fn, label in pbar: - pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name)) img = cv.imread(fn) img = cv.cvtColor(img, cv.COLOR_BGR2RGB)