diff --git a/ppstructure/paddlestructure.py b/ppstructure/paddlestructure.py index c2db42c15fa05e68d07267c002c84f8045c64376..7c25f9c02777013353cdf1535056da84f379bb81 100644 --- a/ppstructure/paddlestructure.py +++ b/ppstructure/paddlestructure.py @@ -146,16 +146,3 @@ def main(): logger.info(item['res']) save_res(result, save_folder, img_name) logger.info('result save to {}'.format(os.path.join(save_folder, img_name))) - - - -if __name__ == '__main__': - table_engine = PaddleStructure(det_model_dir='../inference/table/ch_ppocr_mobile_v2.0_table_det_infer', - rec_model_dir='../inference/table/ch_ppocr_mobile_v2.0_table_rec_infer', - structure_model_dir='../inference/table/ch_ppocr_mobile_v2.0_table_structure_infer', - output='/Users/zhoujun20/Desktop/工作相关/table/table_pr/PaddleOCR/output/table', - show_log=True) - img = cv2.imread('/Users/zhoujun20/Desktop/工作相关/table/table_pr/PaddleOCR/ppstructure/test_imgs/table_1.png') - result = table_engine(img) - for line in result: - print(line)