diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 08441d054d6bc5f2cabccb882f67047b7194899e..98116afd9a21f297d429c26fc88881986c628866 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -268,7 +268,7 @@ class TextDetector(object): post_result = self.postprocess_op(preds, shape_list) dt_boxes = post_result[0]['points'] - print("det_boxes", dt_boxes) + if (self.det_algorithm == "SAST" and self.det_sast_polygon) or ( self.det_algorithm in ["PSE", "FCE", "CT"] and self.postprocess_op.box_type == 'poly'):