test_doc_preprocessor.py 375 B

123456789101112131415
  1. from paddlex import create_pipeline
  2. pipeline = create_pipeline(pipeline="doc_preprocessor")
  3. test_img_path = "./test_imgs/img_rot180_demo.jpg"
  4. # test_img_path = "./test_imgs/doc_distort_test.jpg"
  5. output = pipeline.predict(test_img_path,
  6. use_doc_orientation_classify=True,
  7. use_doc_unwarping=True)
  8. for res in output:
  9. print(res)
  10. res.save_to_img("./output")