将张量数组中的图像保存为 jpg 或 png

Question

我正在尝试使用 mtcnn 检测人脸。 主要目的是检测人脸、裁剪并将裁剪后的图像保存为 jpg 或 png 文件类型。 实现的代码如下。

from facenet_pytorch import MTCNN
from PIL import Image
import numpy as np
from matplotlib import pyplot as plt
img = Image.open("example.jpg")
mtcnn = MTCNN(margin=20, keep_all=True, post_process=False)
faces = mtcnn(img)
print(faces.shape)

这给出了形状

torch.Size([1, 3, 160, 160])

如何将此裁剪部分保存为 jpg 文件。

Answer 1

torch.save(faces, "faces.torch")

那不会被保存为图像，如果你想将它保存为图像：

img = Image.fromarray(faces.cpu().detach().numpy()[0])
img.save("faces.png")