python 从 dict 中保存 image

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