add elapse to cls

fa2d08f2 · LDOUBLEV · b8834933 · fa2d08f2
Commit fa2d08f2 authored Jun 28, 2021 by LDOUBLEV
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tools/infer/predict_cls.py tools/infer/predict_cls.py +1 -0

No files found.
--- 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