在 Python 中使用 imghdr 模块时出现 IOError

Question

这是我用来检查存储在许多文件夹中的文件格式的代码：

check_image_format():

import imghdr as ih

def check_image_format(image_dir):
    for root, dirs, files in os.walk(image_dir):
        for image in files:
            format = ih.what(image)
            if format != 'jpeg' or format != 'png':
                return -1
    return 0

main():

def main(_):
    # Check the correct format of images
    ret = check_image_format('img_dir')
    if(ret == -1):
         print("Some images are not in the correct format. Please check")

my img_dir 是其他三个子文件夹的根目录，其中包含我要查看的图像。当我启动程序时收到此错误：

IOError: [Errno 2] No such file or directory: img_1.jpg

但是文件存在并且在子文件夹中。这个错误的原因是什么？

Answer 1

您需要为当前图片路径建立绝对路径：

import imghdr as ih

def check_image_format(image_dir):

    for root, dirs, files in os.walk(image_dir):
        for image in files:
            format = ih.what(os.path.join(root, image))
            if format != 'jpeg' or format != 'png':
                return -1
    return 0

在 Python 中使用 imghdr 模块时出现 IOError

IOError when using imghdr module in Python

python

imghdr