Numpy 如何将数组图像转换为字节

在图像处理中，一般使用numpy库进行数组形式的图像处理。但是在实际应用中，需要将数组形式的图像转换为字节形式的图像。在本篇文章中，我们将探讨如何将numpy数组图像转化为字节图像。

阅读更多：Numpy 教程

numpy数组图像

numpy数组图像是像素值组成的一个N维数组。在处理图像时，常使用numpy库进行操作，并将图像转换为numpy数组类型。以下是将.jpg类型图像转为numpy数组的示例代码。

import numpy as np
from PIL import Image

img = Image.open('test.jpg')
img_array = np.array(img)
print(img_array.shape)

输出结果是该图像的numpy数组形状（高，宽，通道数）。

字节图像

字节图像是图像的字节数组形式，通常用于图片传输。将numpy数组图像转换为字节图像是图像通信中的一种重要操作。在Python中可以使用Pillow库中的Image对象的tobytes方法将numpy数组图像转换为字节图像。

import numpy as np
from PIL import Image

img = Image.open('test.jpg')
img_array = np.array(img)
img_bytes = img.tobytes()

将numpy数组图像转换后，我们可以将字节图像发送到接收方进行处理和显示。下面是示例代码。

# 发送方
import socket

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect(('target_ip', 9999))
sock.sendall(img_bytes)
sock.close()

# 接收方
import socket
import numpy as np
from PIL import Image

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('localhost', 9999))
sock.listen(1)
conn, addr = sock.accept()

data = bytearray()
while True:
    buf = conn.recv(1024)
    if not buf:
        break
    data += buf

img_bytes = bytes(data)
img_array = np.frombuffer(img_bytes, dtype=np.uint8)
img = Image.fromarray(img_array)
img.show()