我正在使用Python 2.5。使用Python的标准类,我想确定文件的图像大小。
我听说过PIL(Python图像库),但它需要安装才能工作。
如果不使用任何外部库,只使用Python 2.5自己的模块,我如何获得图像的大小?
注意我想支持常见的图像格式,特别是JPG和PNG。
答案 0 :(得分:84)
这是一个python 3脚本,它返回一个元组,其中包含.png,.gif和.jpeg的图像高度和宽度,而不使用任何外部库(即上面引用的Kurt McKee)。应该相对容易将其转移到Python 2。
import struct
import imghdr
def get_image_size(fname):
'''Determine the image type of fhandle and return its size.
from draco'''
with open(fname, 'rb') as fhandle:
head = fhandle.read(24)
if len(head) != 24:
return
if imghdr.what(fname) == 'png':
check = struct.unpack('>i', head[4:8])[0]
if check != 0x0d0a1a0a:
return
width, height = struct.unpack('>ii', head[16:24])
elif imghdr.what(fname) == 'gif':
width, height = struct.unpack('<HH', head[6:10])
elif imghdr.what(fname) == 'jpeg':
try:
fhandle.seek(0) # Read 0xff next
size = 2
ftype = 0
while not 0xc0 <= ftype <= 0xcf:
fhandle.seek(size, 1)
byte = fhandle.read(1)
while ord(byte) == 0xff:
byte = fhandle.read(1)
ftype = ord(byte)
size = struct.unpack('>H', fhandle.read(2))[0] - 2
# We are at a SOFn block
fhandle.seek(1, 1) # Skip `precision' byte.
height, width = struct.unpack('>HH', fhandle.read(4))
except Exception: #IGNORE:W0703
return
else:
return
return width, height
答案 1 :(得分:62)
Kurts的答案需要稍加修改才能适合我。
首先,在ubuntu上:sudo apt-get install python-imaging
然后:
from PIL import Image
im=Image.open(filepath)
im.size # (width,height) tuple
查看handbook了解详情。
答案 2 :(得分:19)
虽然可以调用open(filename, 'rb')
并通过二进制图像标题查看维度,但安装PIL并花时间编写出色的新软件似乎更有用!您可以获得更好的文件格式支持以及广泛使用带来的可靠性。 From the PIL documentation,您完成任务所需的代码似乎是:
from PIL import Image
im = Image.open('filename.png')
print 'width: %d - height: %d' % im.size # returns (width, height) tuple
至于自己编写代码,我不知道Python标准库中的一个模块可以做你想做的事情。您必须以二进制模式open()
图像并自行开始解码。您可以在以下网址阅读有关格式的信息:
答案 3 :(得分:18)
这是一种无需第三方模块即可获取png文件尺寸的方法。来自http://coreygoldberg.blogspot.com/2013/01/python-verify-png-file-and-get-image.html
import struct
def get_image_info(data):
if is_png(data):
w, h = struct.unpack('>LL', data[16:24])
width = int(w)
height = int(h)
else:
raise Exception('not a png image')
return width, height
def is_png(data):
return (data[:8] == '\211PNG\r\n\032\n'and (data[12:16] == 'IHDR'))
if __name__ == '__main__':
with open('foo.png', 'rb') as f:
data = f.read()
print is_png(data)
print get_image_info(data)
当你运行它时,它将返回:
True
(x, y)
另一个例子包括处理JPEG: http://markasread.net/post/17551554979/get-image-size-info-using-pure-python-code
答案 4 :(得分:4)
如果您安装了ImageMagick,则可以使用“identify”。例如,您可以这样调用它:
path = "//folder/image.jpg"
dim = subprocess.Popen(["identify","-format","\"%w,%h\"",path], stdout=subprocess.PIPE).communicate()[0]
(width, height) = [ int(x) for x in re.sub('[\t\r\n"]', '', dim).split(',') ]
答案 5 :(得分:3)
关于Fred the Fantastic's answer:
并非C0
- CF
之间的每个JPEG标记都是SOF
标记;我排除了DHT(C4
),DNL(C8
)和DAC(CC
)。请注意,我还没有考虑是否有可能以这种方式解析C0
和C2
以外的任何帧。但是,其他的似乎相当罕见(除了C0
和C2
之外,我个人还没有遇到过。
无论哪种方式,这都解决了 Malandy 注释中提到的问题Bangles.jpg
(DHT被错误地解析为SOF)。
1431588037-WgsI3vK.jpg
提到的另一个问题是由于imghdr
只能检测到APP0(EXIF)和APP1(JFIF)标题。
这可以通过向imghdr添加更宽松的测试(例如,简单FFD8
或者FFD8FF
?)或更复杂的事情(甚至可能是数据验证)来解决。通过更复杂的方法,我只发现了以下问题:APP14(FFEE
)(Adobe);第一个标记是DQT(FFDB
);和APP2和issues with embedded ICC_PROFILEs。
下面的修改后的代码也略微改变了对imghdr.what()
的调用:
import struct
import imghdr
def test_jpeg(h, f):
# SOI APP2 + ICC_PROFILE
if h[0:4] == '\xff\xd8\xff\xe2' and h[6:17] == b'ICC_PROFILE':
print "A"
return 'jpeg'
# SOI APP14 + Adobe
if h[0:4] == '\xff\xd8\xff\xee' and h[6:11] == b'Adobe':
return 'jpeg'
# SOI DQT
if h[0:4] == '\xff\xd8\xff\xdb':
return 'jpeg'
imghdr.tests.append(test_jpeg)
def get_image_size(fname):
'''Determine the image type of fhandle and return its size.
from draco'''
with open(fname, 'rb') as fhandle:
head = fhandle.read(24)
if len(head) != 24:
return
what = imghdr.what(None, head)
if what == 'png':
check = struct.unpack('>i', head[4:8])[0]
if check != 0x0d0a1a0a:
return
width, height = struct.unpack('>ii', head[16:24])
elif what == 'gif':
width, height = struct.unpack('<HH', head[6:10])
elif what == 'jpeg':
try:
fhandle.seek(0) # Read 0xff next
size = 2
ftype = 0
while not 0xc0 <= ftype <= 0xcf or ftype in (0xc4, 0xc8, 0xcc):
fhandle.seek(size, 1)
byte = fhandle.read(1)
while ord(byte) == 0xff:
byte = fhandle.read(1)
ftype = ord(byte)
size = struct.unpack('>H', fhandle.read(2))[0] - 2
# We are at a SOFn block
fhandle.seek(1, 1) # Skip `precision' byte.
height, width = struct.unpack('>HH', fhandle.read(4))
except Exception: #IGNORE:W0703
return
else:
return
return width, height
注意:创建完整答案而不是评论,因为我还没有被允许。
答案 6 :(得分:1)
该代码确实完成了两件事:
获取图片尺寸
查找jpg文件的真实EOF
当谷歌搜索时,我对后者更感兴趣。 任务是从数据流中删除jpg文件。因为我没有找到任何方法来使用Pythons的“图像”以获得如此jpg文件的EOF而构成了这个。
此示例中有趣的内容/更改/注释:
使用方法uInt16扩展普通的Python文件类 使源代码更易读和可维护。 使用struct.unpack()搞乱使代码看起来很难看
用“搜寻”替换了“无趣的”区域/块
只是想获得尺寸 你可以删除该行:
hasChunk = ord(byte) not in range( 0xD0, 0xDA) + [0x00]
- &gt;因为只有在阅读图像数据块时才会变得很重要
中的评论#break
在找到尺寸后立即停止阅读。 ......但微笑我所说的 - 你是编码员;)
import struct
import io,os
class myFile(file):
def byte( self ):
return file.read( self, 1);
def uInt16( self ):
tmp = file.read( self, 2)
return struct.unpack( ">H", tmp )[0];
jpeg = myFile('grafx_ui.s00_\\08521678_Unknown.jpg', 'rb')
try:
height = -1
width = -1
EOI = -1
type_check = jpeg.read(2)
if type_check != b'\xff\xd8':
print("Not a JPG")
else:
byte = jpeg.byte()
while byte != b"":
while byte != b'\xff': byte = jpeg.byte()
while byte == b'\xff': byte = jpeg.byte()
# FF D8 SOI Start of Image
# FF D0..7 RST DRI Define Restart Interval inside CompressedData
# FF 00 Masked FF inside CompressedData
# FF D9 EOI End of Image
# http://en.wikipedia.org/wiki/JPEG#Syntax_and_structure
hasChunk = ord(byte) not in range( 0xD0, 0xDA) + [0x00]
if hasChunk:
ChunkSize = jpeg.uInt16() - 2
ChunkOffset = jpeg.tell()
Next_ChunkOffset = ChunkOffset + ChunkSize
# Find bytes \xFF \xC0..C3 That marks the Start of Frame
if (byte >= b'\xC0' and byte <= b'\xC3'):
# Found SOF1..3 data chunk - Read it and quit
jpeg.seek(1, os.SEEK_CUR)
h = jpeg.uInt16()
w = jpeg.uInt16()
#break
elif (byte == b'\xD9'):
# Found End of Image
EOI = jpeg.tell()
break
else:
# Seek to next data chunk
print "Pos: %.4x %x" % (jpeg.tell(), ChunkSize)
if hasChunk:
jpeg.seek(Next_ChunkOffset)
byte = jpeg.byte()
width = int(w)
height = int(h)
print("Width: %s, Height: %s JpgFileDataSize: %x" % (width, height, EOI))
finally:
jpeg.close()
答案 7 :(得分:1)
在另一个Stackoverflow帖子中找到了一个不错的解决方案(仅使用标准库+处理jpg):JohnTESlade answer
另一种解决方案(快捷方式)适合那些能够负担得起运行&#39; 档案&#39; python中的命令,运行:
import os
info = os.popen("file foo.jpg").read()
print info
<强>输出强>:
foo.jpg: JPEG image data...density 28x28, segment length 16, baseline, precision 8, 352x198, frames 3
您现在要做的就是格式化输出以捕获尺寸。 352x198 就我而言。
答案 8 :(得分:0)
这取决于文件的输出,我不确定该文件是否在所有系统上都是标准化的。某些JPEG无法报告图片大小
import subprocess, re
image_size = list(map(int, re.findall('(\d+)x(\d+)', subprocess.getoutput("file" + filename))[-1]))
答案 9 :(得分:-2)
偶然发现了这个,但只要你导入numpy就可以使用以下内容。
import numpy as np
[y, x] = np.shape(img[:,:,0])
它的作用是因为你忽略了除一种颜色之外的所有颜色,然后图像只是2D,所以形状告诉你它的出价。仍然是Python的新手,但似乎是一种简单的方法。