如何将 torch.tensor 转换为 base64 图像？

【问题标题】：How to convert torch.tensor to base64 image?如何将 torch.tensor 转换为 base64 图像？
【发布时间】：2023-01-26 18:38:23
【问题描述】：

这是我的张量：

import torch
from torchvision import transforms

content:
tensor([[[[0.8939, 0.8700, 0.8458,  ..., 0.7610, 0.7093, 0.6909],
                                     .
                                     .
                                     .
          [0.4880, 0.5192, 0.5957,  ..., 0.8569, 0.9148, 0.9186]]]])

我想将这个 torch.Tensor 转换为 base64。我已经试过了，但我也无法将 PILImage 转换为 base64。

img = transforms.ToPILImage(content)

我怎样才能做到这一点？

【问题讨论】：

你的张量有多大？
torch.Size([1, 3, 512, 512])

标签： pytorch base64

【解决方案1】：

您应该先将图像转换为字节，然后再将图像编码为 base64 图像

import torch
from torchvision import transforms
import base64

tensor = torch.randn(3, 512, 512)
transform = transforms.ToPILImage()
pil_image = transform(tensor)

# Convert the PIL image to bytes
image_bytes = pil_image.tobytes()

# Encode the bytes to base64
base64_string = base64.b64encode(image_bytes).decode()
print(base64_string)

输出：

00Hbhc4yZEZnw/VlIyHF6OH2VRAXeRdYiH82tBgSBoocDAB+cycknk3c5G9ZmdbHNu90whgDK49cjELZ6uJCzqYxVclXbJRc0Nb2AgN9XWULFOK+ubWJAyNbAgNZRvmwocwJN4603

【讨论】：