python - 需要一个类似字节的对象，而不是 'Image'

Question

我想将 Image 对象写入磁盘，但我不断收到错误:

 a bytes-like object is required, not 'Image'

首先，我将字符串转换为数组，然后使用该数组创建图像。

    class Item(object):

def __init__(self, patch, coords, label):

    self.channels = patch.shape[2]
    # Assuming only square images.
    self.size = patch.shape[0]
    self.data = patch.tobytes()
    self.label = label # Integer label ie, 2 = Carcinoma in situ
    self.coords = coords

def get_label_array(self, num_classes):
    l = np.zeros((num_classes))
    l[self.label] = 1
    return l

def get_patch(self):
    return np.fromstring(self.data, dtype=np.uint8).reshape(self.size, self.size, self.channels)

def get_patch_as_image(self):
    return Image.fromarray(self.get_patch(), 'RGB')

有什么方法可以使用以下方式保存图像:

def save_in_disk(patches, coords, file_name, labels=[]):
    use_label = False
    if len(labels) > 0:
        use_label = True

    # txn is a Transaction object
    for i in range(len(patches)):
        if use_label:
            item = Item(patches[i], coords[i], labels[i])
        else:
            item = Item(patches[i], coords[i], 0)

        p = item.get_patch_as_image()

        str_id = file_name + '-' + str(coords[i][0]) + '-' + str(coords[i][1]) + '.png'
        print(str_id)
        with open(str_id, 'wb+') as f:
            f.write(p)

知道出了什么问题吗？

问候

Answer 1

你使用什么库？

如果你想将 png 图像文件写入磁盘，你需要获取格式化数据，例如使用 BytesIO:

from io import BytesIO
from PIL import Image, ImageDraw

image = Image.new("RGB", (300, 50))
draw = ImageDraw.Draw(image)
draw.text((0, 0), "This text is drawn on image")

byte_io = BytesIO()

image.save(byte_io, 'PNG')

根据您使用的库的不同，它会有所不同。

<小时/>

如果要将Python图像对象保存到磁盘，可以使用pickle/序列化。

如果是纯 Python 类，您可以简单地使用 pickle:

import pickle
with open('Storage', 'wb') as f:
    pickle.dump(instance001, f)

and load it:

with open('Storage', 'rb') as f:
    instance002 = pickle.load(f)

print(instance002.a)   # 2
print(instance002.b)   # 200

<小时/>

看来您使用的是 PIL。将图像另存为 PNG 文件，如下所示:

newImg1.save("img1.png","PNG")