small fix

Author: HaoyangLee

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)

tools/infer/text/predict_system.py

@@ -97,6 +97,8 @@ 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()