未验证 提交 d7232530 编写于 作者: L Labib Asari 提交者: GitHub

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

上级 578fa8e4
......@@ -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)
......
......@@ -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]])
......
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册