Python的PIL：查找图像的大小，而不写为文件

Question

编辑：这个问题已经被标记为重复？我的问题显然是关于优化这个过程，而不是如何去做。我甚至提供了代码来证明我已经知道后者。在互联网大厅监视器上标记它们之前，甚至在标题之前读过这些问题吗？

我有下面的代码块来压缩用PIL的图像，直到所述图像是具有一定规模下。

from PIL import Image 
import os 

def compress(image_file, max_size, scale): 
    while os.path.getsize(image_file) > max_size: 
     pic = Image.open(image_file) 
     original_size = pic.size 
     pic = pic.resize((int(original_size[0] * scale), 
      int(original_size[1] * scale)), 
      Image.ANTIALIAS) 
     pic.save(image_file, optimize=True, quality=95) 

在这段代码中，我使用os.path.getsize(image_file)来获得图像的大小。然而，这意味着文件必须保存在每次循环运行pic.save(image_file, optimize=True, quality=95。

这个过程需要很长的时间。

有没有办法通过某种方式来获取图像的大小PILImage对象pic？

Answer 1

使用io.BytesIO()将图像保存到内存中。它也可能是更好的从原始文件中的每个时间进行如下调整：

from PIL import Image 
import os 
import io 

def compress(original_file, max_size, scale): 
    assert(0.0 < scale < 1.0) 
    orig_image = Image.open(original_file) 
    cur_size = orig_image.size 

    while True: 
     cur_size = (int(cur_size[0] * scale), int(cur_size[1] * scale)) 
     resized_file = orig_image.resize(cur_size, Image.ANTIALIAS) 

     with io.BytesIO() as file_bytes: 
      resized_file.save(file_bytes, optimize=True, quality=95, format='jpeg') 

      if file_bytes.tell() <= max_size: 
       file_bytes.seek(0, 0) 
       with open(original_file, 'wb') as f_output: 
        f_output.write(file_bytes.read()) 
       break 

compress(r"c:\mytest.jpg", 10240, 0.9) 

因此，这将需要该文件，直到达到一个合适的大小规模下来0.9每次尝试。然后它会覆盖原始文件。

---

作为一个替代方法，你可以创建秤的列表尝试，例如[0.01, 0.02 .... 0.99, 1]然后使用二进制印章，以确定哪些规模的结果在最近的一个文件大小来max_size如下：

def compress(original_file, max_size): 
    save_opts={'optimize':True, 'quality':95, 'format':'jpeg'} 
    orig_image = Image.open(original_file) 
    width, height = orig_image.size 
    scales = [scale/1000 for scale in range(1, 1001)] # e.g. [0.001, 0.002 ... 1.0] 

    lo = 0 
    hi = len(scales) 

    while lo < hi: 
     mid = (lo + hi) // 2 

     scaled_size = (int(width * scales[mid]), int(height * scales[mid])) 
     resized_file = orig_image.resize(scaled_size, Image.ANTIALIAS) 

     file_bytes = io.BytesIO() 
     resized_file.save(file_bytes, **save_opts) 
     size = file_bytes.tell() 
     print(size, scales[mid]) 

     if size < max_size: 
      lo = mid + 1 
     else: 
      hi = mid 

    scale = scales[max(0, lo-1)] 
    print("Using scale:", scale) 
    orig_image.resize((int(width * scale), int(height * scale)), Image.ANTIALIAS).save(original_file, **save_opts) 

所以对于10000一个max_size，循环首先尝试的0.501规模，如果过大0.251受审等等。当max_size=1024将尝试以下比例：

180287 0.501 
56945 0.251 
17751 0.126 
5371 0.063 
10584 0.095 
7690 0.079 
9018 0.087 
10140 0.091 
9336 0.089 
9948 0.09 
Using scale: 0.09 

Answer 2

可能的话，你可以使用StringIO的在内存中执行的操作在this answer看看。

import StringIO 

output = StringIO.StringIO() 
image.save(output) 
contents = output.getvalue() 
output.close() 

image.save(output, format="GIF")