使用 pdf2image 包将 pdf 转换为图像后无法使用 cv2 调整图像大小

Question

使用 pdf2image 包将 pdf 转换为图像后，无法使用 cv2 调整图像大小。

代码是

import os
import cv2
from pdf2image import convert_from_path
pdf_files = [filename for filename in os.listdir(
    os.getcwd()) if filename.endswith('.pdf')]
for pdf_file in pdf_files:
    images = convert_from_path(pdf_file, 400)
    for i, image in enumerate(images):
        fname = pdf_file+'_image'+str(i)+'.jpg'
        fname = cv2.resize(fname, (3400, 4400))
        image.save(fname, "JPEG")

错误是

fname = cv2.resize(fname, (3400, 4400))
TypeError: Expected Ptr<cv::UMat> for argument 'src'

Answer 1

您正在尝试在这行代码中调整文件名的大小：

fname = cv2.resize(fname, (3400, 4400))

将上面的行替换为：

image = cv2.resize(image, (3400, 4400))

要保存图像，命令将是cv2.imwrite(fname, image)

您还需要将图像转换为数组。为此，您必须import numpy。完整代码如下：

import os
import cv2
import numpy as np 
from pdf2image import convert_from_path
pdf_files = [filename for filename in os.listdir(
    os.getcwd()) if filename.endswith('.pdf')]
for pdf_file in pdf_files:
    images = convert_from_path(pdf_file, 400)
    for i, image in enumerate(images):
        fname = pdf_file+'_image'+str(i)+'.jpg'
        image = np.array(image)
        image = cv2.resize(image, (3400, 4400))
        cv2.imwrite(fname, image)