mirror of
https://github.com/Zeyi-Lin/HivisionIDPhotos.git
synced 2025-09-15 14:58:34 +08:00
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
from PIL import Image
|
||
import io
|
||
import numpy as np
|
||
|
||
|
||
def resize_image_to_kb(input_image, output_image_path, target_size_kb):
|
||
"""
|
||
Resize an image to a target size in KB.
|
||
将图像调整大小至目标文件大小（KB）。
|
||
|
||
:param input_image_path: Path to the input image. 输入图像的路径。
|
||
:param output_image_path: Path to save the resized image. 保存调整大小后的图像的路径。
|
||
:param target_size_kb: Target size in KB. 目标文件大小（KB）。
|
||
|
||
Example:
|
||
resize_image_to_kb('input_image.jpg', 'output_image.jpg', 50)
|
||
"""
|
||
|
||
if isinstance(input_image, np.ndarray):
|
||
img = Image.fromarray(input_image)
|
||
elif isinstance(input_image, Image.Image):
|
||
img = input_image
|
||
else:
|
||
raise ValueError("input_image must be a NumPy array or PIL Image.")
|
||
|
||
# Convert image to RGB mode if it's not
|
||
if img.mode != "RGB":
|
||
img = img.convert("RGB")
|
||
|
||
# Initial quality value
|
||
quality = 95
|
||
|
||
while True:
|
||
# Create a BytesIO object to hold the image data in memory
|
||
img_byte_arr = io.BytesIO()
|
||
|
||
# Save the image to the BytesIO object with the current quality
|
||
img.save(img_byte_arr, format="JPEG", quality=quality)
|
||
|
||
# Get the size of the image in KB
|
||
img_size_kb = len(img_byte_arr.getvalue()) / 1024
|
||
|
||
# Check if the image size is within the target size
|
||
if img_size_kb <= target_size_kb or quality == 1:
|
||
# If the image is smaller than the target size, add padding
|
||
if img_size_kb < target_size_kb:
|
||
padding_size = int(
|
||
(target_size_kb * 1024) - len(img_byte_arr.getvalue())
|
||
)
|
||
padding = b"\x00" * padding_size
|
||
img_byte_arr.write(padding)
|
||
|
||
# Save the image to the output path
|
||
with open(output_image_path, "wb") as f:
|
||
f.write(img_byte_arr.getvalue())
|
||
break
|
||
|
||
# Reduce the quality if the image is still too large
|
||
quality -= 5
|
||
|
||
# Ensure quality does not go below 1
|
||
if quality < 1:
|
||
quality = 1
|