fix: fixed issue #544 (#545)

Author: SWHL

python/rapidocr/ch_ppocr_cls/main.py

@@ -34,6 +34,8 @@ def __init__(self, cfg: Dict[str, Any]):
         self.session = get_engine(cfg.engine_type)(cfg)
 
     def __call__(self, img_list: Union[np.ndarray, List[np.ndarray]]) -> TextClsOutput:
+        start_time = time.perf_counter()
+
         if isinstance(img_list, np.ndarray):
             img_list = [img_list]
 
@@ -59,17 +61,17 @@ def __call__(self, img_list: Union[np.ndarray, List[np.ndarray]]) -> TextClsOutp
                 norm_img_batch.append(norm_img)
             norm_img_batch = np.concatenate(norm_img_batch).astype(np.float32)
 
-            starttime = time.time()
             prob_out = self.session(norm_img_batch)
             cls_result = self.postprocess_op(prob_out)
-            elapse += time.time() - starttime
 
             for rno, (label, score) in enumerate(cls_result):
                 cls_res[indices[beg_img_no + rno]] = (label, score)
                 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.perf_counter() - start_time
         return TextClsOutput(img_list=img_list, cls_res=cls_res, elapse=elapse)
 
     def resize_norm_img(self, img: np.ndarray) -> np.ndarray:

python/rapidocr/ch_ppocr_rec/main.py

@@ -84,6 +84,8 @@ def get_character_dict(self, cfg):
         return character, dict_path
 
     def __call__(self, args: TextRecInput) -> TextRecOutput:
+        start_time = time.perf_counter()
+
         img_list = [args.img] if isinstance(args.img, np.ndarray) else args.img
         return_word_box = args.return_word_box
 
@@ -116,7 +118,6 @@ def __call__(self, args: TextRecInput) -> TextRecOutput:
                 norm_img_batch.append(norm_img[np.newaxis, :])
             norm_img_batch = np.concatenate(norm_img_batch).astype(np.float32)
 
-            start_time = time.perf_counter()
             preds = self.session(norm_img_batch)
             line_results, word_results = self.postprocess_op(
                 preds,
@@ -131,10 +132,11 @@ def __call__(self, args: TextRecInput) -> TextRecOutput:
                     continue
 
                 rec_res[indices[beg_img_no + rno]] = (one_res, None)
-            elapse += time.perf_counter() - start_time
 
         all_line_results, all_word_results = list(zip(*rec_res))
         txts, scores = list(zip(*all_line_results))
+
+        elapse = time.perf_counter() - start_time
         return TextRecOutput(
             img_list,
             txts,