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
python 从 dict 中保存 image
最新推荐文章于 2024-08-10 23:14:03 发布