add elapse to cls

2021-06-28 13:50:29 +08:00 · 2021-06-28 13:50:29 +08:00 · fa2d08f211
parent b88349335b
commit fa2d08f211
1 changed files with 1 additions and 0 deletions
--- 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