如何把torch.tensor转换成base64图片？-解网

问：

这是我的张量：

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]]]])

我想转换这个火炬。张量到 base64。我已经尝试过了，但我也无法将 PILImage 转换为 base64。

img = transforms.ToPILImage(content)

我该怎么做？

pytorch 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

如何把torch.tensor转换成base64图片？

How to convert torch.tensor to base64 image?

评论

评论