test_layout_parsing.py 613 B

12345678910111213141516171819
  1. from paddlex import create_pipeline
  2. pipeline = create_pipeline(pipeline="layout_parsing")
  3. output = pipeline.predict("./test_imgs/test_layout_parsing.jpg",
  4. use_doc_orientation_classify=True,
  5. use_doc_unwarping=True,
  6. use_common_ocr=True,
  7. use_seal_recognition=True,
  8. use_table_recognition=True)
  9. # output = pipeline("./test_imgs/demo_paper.png")
  10. # output = pipeline("./test_imgs/table_recognition.jpg")
  11. # output = pipeline.predict("./test_imgs/seal_text_det.png")
  12. # output = pipeline.predict("./test_imgs/img_rot180_demo.jpg")
  13. for res in output:
  14. # print(res)
  15. res.save_results("./output")