Skip to content

Commit d723253

Browse files
authored
move setting description outside for loop in evaluation scripts (#134)
1 parent 578fa8e commit d723253

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

tools/eval/datasets/icdar.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,9 @@ def load_label(self, label_file):
3333
def eval(self, model):
3434
right_num = 0
3535
pbar = tqdm(self.val_label)
36+
pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
37+
3638
for fn, label in pbar:
37-
pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
3839

3940
img = cv.imread(fn)
4041

tools/eval/datasets/iiit5k.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,9 @@ def load_label(self, root):
3636
def eval(self, model):
3737
right_num = 0
3838
pbar = tqdm(self.val_label)
39+
pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
40+
3941
for img, value in pbar:
40-
pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
4142

4243

4344
rbbox = np.array([0, img.shape[0], 0, 0, img.shape[1], 0, img.shape[1], img.shape[0]])

tools/eval/datasets/imagenet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,9 @@ def eval(self, model):
3939
top_1_hits = 0
4040
top_5_hits = 0
4141
pbar = tqdm(self.val_label)
42+
pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
43+
4244
for fn, label in pbar:
43-
pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
4445

4546
img = cv.imread(fn)
4647
img = cv.cvtColor(img, cv.COLOR_BGR2RGB)

0 commit comments

Comments
 (0)