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)
|