def save_image(img,save_dir,name): if type(img).__module__ == np.__name__: PIL = Image.fromarray(img.astype(np.uint8)) else: PIL = img PIL.save(save_dir+name+".png")