From fa2d08f211a8f5f9e89e103df64682918d509f2d Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Mon, 28 Jun 2021 13:50:29 +0800
Subject: [PATCH] add elapse to cls

---
 tools/infer/predict_cls.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tools/infer/predict_cls.py b/tools/infer/predict_cls.py
index 8f2a1361..1a0cf356 100755
--- a/tools/infer/predict_cls.py
+++ b/tools/infer/predict_cls.py
@@ -112,6 +112,7 @@ class TextClassifier(object):
                 if '180' in label and score > self.cls_thresh:
                     img_list[indices[beg_img_no + rno]] = cv2.rotate(
                         img_list[indices[beg_img_no + rno]], 1)
+        elapse = time.time() - starttime
         return img_list, cls_res, elapse
 
 
-- 
GitLab