diff --git a/PP-LCNet_x1_0_doc_ori_infer.onnx b/PP-LCNet_x1_0_doc_ori_infer.onnx new file mode 100644 index 0000000..4a95db1 Binary files /dev/null and b/PP-LCNet_x1_0_doc_ori_infer.onnx differ diff --git a/rapid_table_det/predictor.py b/rapid_table_det/predictor.py index 0c61de1..0d66991 100644 --- a/rapid_table_det/predictor.py +++ b/rapid_table_det/predictor.py @@ -296,7 +296,8 @@ class PPLCNet: def __init__(self, config: Dict[str, Any]): self.model = OrtInferSession(config) self.img_loader = LoadImage() - self.resize_shape = [624, 624] + # self.resize_shape = [624, 624] + self.resize_shape = [224, 224] def __call__(self, img, **kwargs): start = time.time() @@ -311,7 +312,8 @@ def __call__(self, img, **kwargs): pred_label = mini_batch_result[0] return pred_label, time.time() - start - def img_preprocess(self, img, resize_shape=[624, 624]): + #def img_preprocess(self, img, resize_shape=[624, 624]): + def img_preprocess(self, img, resize_shape=[224, 224]): im, new_w, new_h, left, top = ResizePad(img, resize_shape[0]) im = np.array(im).transpose((2, 0, 1)) / 255.0 return im[None, :].astype("float32")