import os
from PIL import Image


def compress_under_size(imagefile, targetfile, targetsize):
    """壓縮圖片尺寸直到某一尺寸

    :param imagefile: 原圖路徑
    :param targetfile: 保存圖片路徑
    :param targetsize: 目標(biāo)大小，單位byte
    """
    currentsize = os.path.getsize(imagefile)
    for quality in range(99, 0, -1):  # 壓縮質(zhì)量遞減
        if currentsize > targetsize:
            image = Image.open(imagefile)
            image.save(targetfile, optimize=True, quality=quality)
            currentsize = os.path.getsize(targetfile)


if __name__ == '__main__':
    imagefile = '1.jpg'  # 圖片路徑
    targetfile = 'result.jpg'  # 目標(biāo)圖片路徑
    targetsize = 2 * 1024  # 目標(biāo)圖片大小
    compress_under_size(imagefile, targetfile, targetsize)  # 將圖片壓縮到2KB

效果

注意！無(wú)法實(shí)現(xiàn)圖片無(wú)限壓縮，若文件太小，辨識(shí)度也會(huì)大大降低

2. 減小圖片尺寸

import os
from PIL import Image


def image_compress(filename, savename, targetsize):
    """圖像壓縮

    :param filename: 原圖路徑
    :param savename: 保存圖片路徑
    :param targetsize: 目標(biāo)大小，單位為byte
    """
    image = Image.open(filename)
    size = os.path.getsize(filename)
    if size <= targetsize:
        return
    width, height = image.size
    num = (targetsize / size) ** 0.5
    width, height = round(width * num), round(height * num)
    image.resize((width, height)).save(savename)


if __name__ == '__main__':
    filename = '1.jpg'
    savename = 'result.jpg'
    targetsize = 2 * 1024
    image_compress(filename, savename, targetsize)

效果

增加文件大小

Windows

通過 subprocess 模塊調(diào)用系統(tǒng)命令 fsutil file createnew filename filesize 創(chuàng)建指定大小的文件

再用 copy/b 命令合并數(shù)據(jù)到圖片上

import os
import time
import subprocess

imagefile = '1.jpg'  # 圖片路徑
targetfile = 'result.jpg'  # 目標(biāo)圖片路徑
targetsize = 10 * 1024 * 1024  # 目標(biāo)圖片大小

tempfile = str(int(time.time()))  # 臨時(shí)文件路徑
tempsize = str(targetsize - os.path.getsize(imagefile))  # 臨時(shí)文件大小
subprocess.run(['fsutil', 'file', 'createnew', tempfile, tempsize])  # 創(chuàng)建臨時(shí)文件
subprocess.run(['copy/b', '{}/b+{}/b'.format(imagefile, tempfile), targetfile], shell=True)  # 合并生成新圖片
os.remove(tempfile)

Linux

通過 subprocess 模塊調(diào)用系統(tǒng)命令 fallocate -l filesize filename 創(chuàng)建指定大小的文件

再用 cat > 命令合并數(shù)據(jù)到圖片上

import os
import time
import subprocess

imagefile = '1.jpg'  # 圖片路徑
targetfile = 'result.jpg'  # 目標(biāo)圖片路徑
targetsize = 10 * 1024 * 1024  # 目標(biāo)圖片大小

tempfile = str(int(time.time()))  # 臨時(shí)文件路徑
tempsize = str(targetsize - os.path.getsize(imagefile))  # 臨時(shí)文件大小
subprocess.run(['fallocate', '-l', tempsize, tempfile])  # 創(chuàng)建臨時(shí)文件
subprocess.run('cat {} {} > {}'.format(imagefile, tempfile, targetfile), shell=True)  # 合并生成新圖片
os.remove(tempfile)

效果

圖片的分辨率沒變

封裝

import os
import time
import platform
import subprocess
from PIL import Image


def resize_picture_filesize(imagefile, targetfile, targetsize):
    """調(diào)整圖片文件大小

    :param imagefile: 原圖路徑
    :param targetfile: 保存圖片路徑
    :param targetsize: 目標(biāo)文件大小，單位byte
    """
    currentsize = os.path.getsize(imagefile)  # 原圖文件大小

    if currentsize > targetsize:  # 需要縮小
        for quality in range(99, 0, -1):  # 壓縮質(zhì)量遞減
            if currentsize > targetsize:
                image = Image.open(imagefile)
                image.save(targetfile, optimize=True, quality=quality)
                currentsize = os.path.getsize(targetfile)
    else:  # 需要放大
        system = platform.system()
        tempfile = str(int(time.time()))  # 臨時(shí)文件路徑
        tempsize = str(targetsize - os.path.getsize(imagefile))  # 臨時(shí)文件大小

        if system == 'Windows':
            subprocess.run(['fsutil', 'file', 'createnew', tempfile, tempsize])  # 創(chuàng)建臨時(shí)文件
            subprocess.run(['copy/b', '{}/b+{}/b'.format(imagefile, tempfile), targetfile], shell=True)  # 合并生成新圖片
        elif system == 'Linux':
            subprocess.run(['fallocate', '-l', tempsize, tempfile])  # 創(chuàng)建臨時(shí)文件
            subprocess.run('cat {} {} > {}'.format(imagefile, tempfile, targetfile), shell=True)  # 合并生成新圖片
        os.remove(tempfile)


if __name__ == '__main__':
    imagefile = '1.jpg'  # 8KB的圖片
    resize_picture_filesize(imagefile, 'reduce.jpg', 2 * 1024)  # 縮小到2KB
    resize_picture_filesize(imagefile, 'increase.jpg', 800 * 1024)  # 放大到800KB