将张量数组中的图像保存为 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")

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

Saving Image in tensor array as jpg or png

python

face-detection

pytorch