Labib Asari commited on
Commit
5de3983
·
1 Parent(s): 12c3d6b

move setting description outside for loop in evaluation scripts (#134)

Browse files
tools/eval/datasets/icdar.py CHANGED
@@ -33,8 +33,9 @@ class ICDAR:
33
  def eval(self, model):
34
  right_num = 0
35
  pbar = tqdm(self.val_label)
 
 
36
  for fn, label in pbar:
37
- pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
38
 
39
  img = cv.imread(fn)
40
 
 
33
  def eval(self, model):
34
  right_num = 0
35
  pbar = tqdm(self.val_label)
36
+ pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
37
+
38
  for fn, label in pbar:
 
39
 
40
  img = cv.imread(fn)
41
 
tools/eval/datasets/iiit5k.py CHANGED
@@ -36,8 +36,9 @@ class IIIT5K:
36
  def eval(self, model):
37
  right_num = 0
38
  pbar = tqdm(self.val_label)
 
 
39
  for img, value in pbar:
40
- pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
41
 
42
 
43
  rbbox = np.array([0, img.shape[0], 0, 0, img.shape[1], 0, img.shape[1], img.shape[0]])
 
36
  def eval(self, model):
37
  right_num = 0
38
  pbar = tqdm(self.val_label)
39
+ pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
40
+
41
  for img, value in pbar:
 
42
 
43
 
44
  rbbox = np.array([0, img.shape[0], 0, 0, img.shape[1], 0, img.shape[1], img.shape[0]])
tools/eval/datasets/imagenet.py CHANGED
@@ -39,8 +39,9 @@ class ImageNet:
39
  top_1_hits = 0
40
  top_5_hits = 0
41
  pbar = tqdm(self.val_label)
 
 
42
  for fn, label in pbar:
43
- pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
44
 
45
  img = cv.imread(fn)
46
  img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
 
39
  top_1_hits = 0
40
  top_5_hits = 0
41
  pbar = tqdm(self.val_label)
42
+ pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
43
+
44
  for fn, label in pbar:
 
45
 
46
  img = cv.imread(fn)
47
  img = cv.cvtColor(img, cv.COLOR_BGR2RGB)