Como se muestra a continuación: lateral_map_2 es un vector de tipo tensor (mapa de predicción) generado por la red, y ahora quiero convertirlo en una imagen, y la prueba personal es efectiva.
from torchvision.utils import save_image
res = lateral_map_2 # [1,1,352,352]
# res = res * 255
res.cuda()
tensor_float = res.float()
img = tensor_float[0]
# (图片,存放图片的路径)
save_image(img, opt.save_path + name)