diff --git a/test1/paddlestructure.py b/test1/paddlestructure.py index d8199101..bbe69d40 100644 --- a/test1/paddlestructure.py +++ b/test1/paddlestructure.py @@ -146,23 +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(show_log=True) - - img_path = '../test/test_imgs/PMC1173095_006_00.png' - img = cv2.imread(img_path) - result = table_engine(img) - save_res(result, '/Users/zhoujun20/Desktop/工作相关/table/table_pr/PaddleOCR/output/table', - os.path.basename(img_path).split('.')[0]) - - for line in result: - print(line) - - from PIL import Image - - font_path = '../doc/fonts/simfang.ttf' - image = Image.open(img_path).convert('RGB') - im_show = draw_result(image, result, font_path=font_path) - im_show = Image.fromarray(im_show) - im_show.save('result.jpg') \ No newline at end of file