Python--保存图片到文件夹

为方便以后开发查阅，特记录此相关代码。
所要实现的功能：

将图片保存于指定文件夹中，可指定图片后缀

代码如下：

def save_img(img, file_name,file_path='img'):
    # Save image to the file_path '\img'
    if not os.path.exists(file_path):
        print (file_path,'not exist')
        os.makedirs(file_path)
    # Image suffix
    # file_suffix = 'jpg'
    # Join image name
    # 项目中给定的图片名为“hiking_125.jpg_0”，需命名切片
    filename = '{}{}{}'.format(file_path, os.sep, file_name[:-2])
    # filename = file_name[:-2]
    # print(filename)
    plt.imsave(filename, img)

附plt常用保存图片的相应代码：
plt.savefig(‘file_name’) 保存带有坐标的图像
plt.imsave(‘file_name’, img) 保存原始图像