python通过图片内容判断图片类型

最新推荐文章于 2024-07-05 03:25:43 发布

shhider

最新推荐文章于 2024-07-05 03:25:43 发布

阅读量2k

点赞数

分类专栏： Python

Python 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种通过读取文件头来判断图片类型的简单方法。该方法支持JPEG、BMP、GIF及PNG等常见格式，并可根据文件路径或文件内容进行识别。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

def get_image_type(pd, is_path=True):
    '''
    获取图片的类型，支持传入路径和文件内容
    '''
    if is_path:
        f = file(pd, 'rb')
        data = f.read(10).encode('hex')
    else:
        data = pd.encode('hex')
 
    ftype = None
 
    if data.startswith('ffd8'):
        ftype = 'jpeg'
    if data.startswith('424d'):
        ftype = 'bmp'
    if data.startswith('474946'):
        ftype = 'gif'
    elif data.startswith('89504e470d0a1a0a'):
        ftype = 'png'
 
    return ftype