瀏覽代碼

fix the iteration method for external_layout_det_results (#4550)

zhang-prog 2 月之前
父節點
當前提交
8e1b52516a
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/pipelines/seal_recognition/pipeline.py

+ 1 - 1
paddlex/inference/pipelines/seal_recognition/pipeline.py

@@ -259,7 +259,7 @@ class _SealRecognitionPipeline(BasePipeline):
                     layout_det_results = []
                     for _ in doc_preprocessor_images:
                         try:
-                            layout_det_res = list(external_layout_det_results)[0]
+                            layout_det_res = next(external_layout_det_results)
                         except StopIteration:
                             raise ValueError("No more layout det results")
                         layout_det_results.append(layout_det_res)