Skip to content

Commit 9a401db

Browse files
authored
readers.py: simplify pypdfium2 use
1 parent caa334a commit 9a401db

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

paddlex/inference/utils/io/readers.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -293,11 +293,7 @@ def read_file(self, in_path):
293293
doc = pdfium.PdfDocument(in_path)
294294
try:
295295
for page in doc:
296-
image = page.render(scale=self._scale, rotation=self._rotation).to_pil()
297-
image = image.convert("RGB")
298-
img_cv = np.array(image)
299-
img_cv = cv2.cvtColor(img_cv, cv2.COLOR_RGB2BGR)
300-
yield img_cv
296+
yield page.render(scale=self._scale, rotation=self._rotation).to_numpy()
301297
finally:
302298
doc.close()
303299

0 commit comments

Comments
 (0)