python 从 dict 中保存 image

howtoloveyou

于 2022-04-23 21:32:32 发布

阅读量1.7k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：语义分割文章标签： python

本文链接：https://blog.youkuaiyun.com/qq_38758049/article/details/124370608

语义分割专栏收录该内容

1 篇文章

订阅专栏

本文介绍了如何使用深度学习模型对图像进行处理，详细步骤包括模型预测、数据类型转换、CPU转numpy、图像数组生成和最终保存为PNG格式，展示了图像输出流程的关键操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

image_outputs = model(image_inputs) # dict['est', 'gt', 'raw']
image_outputs = image_outputs['est'] # dist 1x1920x1024
image_outputs = image_outputs[0] # tensor 1x1920x1024
image_outputs = image_outputs.to("cpu").numpy().astype(np.uint8)[0] # ndarry 1920x1024
image_outputs = Image.fromarray(image_outputs) # img 1920x1024
image_outputs.save("./imgs_dir/" + str(batch_idx + 1).zfill(4) + ".png") # ./imgs_dir/0001.png