Python_npy文件与png图片的格式转换

  npy文件是以数组形式保存图片数据,我们有时再进行训练时,可能需要将其进行图片格式的转换,废话不多说,直接上代码:

import numpy as np
import imageio
import os
from skimage import img_as_ubyte
dest_dir_rgb = "./A"  # 文件存储的路径
dest_dir_gray = "./B"  # 文件存储的路径
picture_rgb = np.load('A.npy')    #使用numpy载入npy文件,里面存放的是RGB图片
picture_gray = np.load('B.npy')    #使用numpy载入npy文件,里面存放的是灰度图片
print(picture_rgb.shape)
print(picture_gray.shape)
# RGB图片归一化处理
picture_rgb = picture_rgb.reshape(picture_rgb.shape[0], 256, 256, 3)
picture_rgb = picture_rgb.astype('float32')
picture_rgb = picture_rgb/255.
# 将图片保存在dest_dir_rgb文件夹下
for i in range(0,m):
	imageio.imwrite(os.path.join(dest_dir_rgb, "{}.png".format(i)),img_as_ubyte(picture_rgb[i]))
# 灰度图片归一化处理
picture_gray = picture_gray.reshape(picture_gray.shape[0], 256, 256, 1)
picture_gray = picture_gray/picture_gray.max()
# 将灰度图片保存在dest_dir_gray文件夹下
for i in range(0,n):
	imageio.imwrite(os.path.join(dest_dir_rgb, "{}.png".format(i)),img_as_ubyte(picture_rgb[i]))

  注意:
  1、先通过print(map.shape)查看数组尺寸,比如说[120][256][256][3],我们知道这个npy文件里保存的是120张256*256大小的RGB图片;
  2、上述代码中的m、n就是指的图片数量;
  3、RGB图片与灰度图片的归一化处理有些许差异。

  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值