gaotingquan 6 months ago
parent
commit
410ebec84b
1 changed files with 3 additions and 1 deletions
  1. 3 1
      paddlex/inference/pipelines/doc_preprocessor/pipeline.py

+ 3 - 1
paddlex/inference/pipelines/doc_preprocessor/pipeline.py

@@ -177,10 +177,12 @@ class _DocPreprocessorPipeline(BasePipeline):
             else:
                 output_imgs = rot_imgs
 
-            for input_path, page_index, image_array, output_img in zip(
+            for input_path, page_index, image_array, angle, rot_img, output_img in zip(
                 batch_data.input_paths,
                 batch_data.page_indexes,
                 image_arrays,
+                angles,
+                rot_imgs,
                 output_imgs,
             ):
                 single_img_res = {