时间戳

首页 > 解决方案 > OpenCV TypeError：Python目录中的迭代文件预期

python - OpenCV TypeError：Python目录中的迭代文件预期

问题描述

我在转换目录中的文件时遇到问题。它给了我一个 TypeError，我不知道如何修复它。下面我将我的代码与输出一起粘贴。

directory = './Input/'
for filename in os.scandir(directory):

    print(filename)
    print(type(filename))

    # image = cv2.imread(filename)
    result = sr.upsample(filename)
    cv2.imwrite("./Output/image.png", result)

我的输出：

<DirEntry '1.png'>
<class 'posix.DirEntry'>
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-14-e9096dfd223e> in <module>()
     28 
     29     # image = cv2.imread(filename)
---> 30     result = sr.upsample(filename)
     31     # cv2.imwrite("./Output/image.png", result)
     32     # else:

TypeError: Expected Ptr<cv::UMat> for argument 'img'

标签： pythondirectorycv2

解决方案

upsample 函数采用图像 Mat，而不是 DirEntry 或路径。您需要先阅读图像。

也许尝试：

directory = './Input/'
for filename in os.scandir(directory):
    if (filename.path.endswith(".jpg") or filename.path.endswith(".png")):
        print(filename.path)
        image = cv2.imread(filename.path)
        result = sr.upsample(image)
        cv2.imwrite("./Output/image.png", result)

推荐阅读