Python对np.ndarray格式的图片进行压缩

蜡笔小新星

于 2024-09-10 15:04:48 发布

阅读量375

点赞数 1

CC 4.0 BY-SA版权

文章标签： python 开发语言学习经验分享

本文链接：https://blog.csdn.net/m0_54490473/article/details/141844317

在Python中，如果你想要将一个np.ndarray格式的图片进行压缩，但不保存到硬盘，你可以使用cv2库（OpenCV）进行图片的压缩。以下是一个简单的例子：

import numpy as np
import cv2
 
# 假设你已经有了一个np.ndarray格式的图片数据
image = np.random.randint(0, 256, (100, 100, 3), dtype=np.uint8)  # 示例图片数据
 
# 使用cv2.imencode将np.ndarray格式的图片压缩为JPEG格式的字节流
# 参数1是压缩参数，参数2是需要压缩的图片
_, compressed_image = cv2.imencode('.jpg', image, [int(cv2.IMWRITE_JPEG_QUALITY), 50])
 
# 此时compressed_image是一个字节流，包含了压缩后的图片数据
# 如果你想保留为np.ndarray格式，可以直接使用numpy的frombuffer方法
compressed_image = np.asarray(bytearray(compressed_image), dtype='uint8')
 
# 如果你想要在内存中保持np.ndarray格式，只需要保留compressed_image变量即可
# 不需要将其转换为其他格式，因为它已经是一个np.ndarray对象了

在这个例子中，我们首先创建了一个示例的np.ndarray图片，然后使用cv2.imencode方法将其压缩为JPEG格式的字节流。压缩参数可以调整，例如调整int(cv2.IMWRITE_JPEG_QUALITY), 50中的50来改变压缩质量。最终，我们将压缩后的字节流重新转换为np.ndarray格式，以保持内存中的数据格式不变。