small fix

mindspore-lab · Jun 21, 2023 · f4d31c5 · f4d31c5
1 parent 2a74874
commit f4d31c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tools/infer/text/predict_cls.py b/tools/infer/text/predict_cls.py
@@ -263,4 +263,4 @@ def save_cls_res(cls_res_all, img_paths, include_score=False, save_path="./cls_r
     save_cls_res(cls_res_all, img_paths, include_score=True, save_path=save_fp)
     # print('All cls res: ', cls_res_all)
     print("Done! Text direction classification results saved in ", save_dir)
-    print("Time cost: ", t, "FPS: ", len(img_paths) / t)
+    print("CLS time: ", t, "FPS: ", len(img_paths) / t)
diff --git a/tools/infer/text/predict_system.py b/tools/infer/text/predict_system.py
@@ -97,6 +97,7 @@ def __call__(self, img_or_path: Union[str, np.ndarray], do_visualize=True):
             cls_start = time()
             _, crops = self.direction_classify(crops)
             time_profile["cls"] = time() - cls_start
+            _logger.info(f"\nCls time: {time_profile['cls']}")
 
         # recognize cropped images
         rs = time()