diff --git a/tools/infer/predict_cls.py b/tools/infer/predict_cls.py index 8f2a1361801b699185bc4f257e33f1d0619310e1..1a0cf3567305f22c07b547bdc56dbdfc5a88d737 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