（4）Python实现npy转jpg

78 阅读 0 评论 52 点赞

我是靠谱客的博主安详小懒虫，最近开发中收集的这篇文章主要介绍（4）Python实现npy转jpg，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

仅为学习交流，侵删！转载请注明出处！

目录
npy为三维，例如：（500，256，256）代表存储500张256x256的灰度图像

import cv2
import numpy as np
depth_dataset = np.load("E:/numpydata/test.npy")
# print(depth_dataset.shape)
#遍历
for i in range(0,depth_dataset.shape[0]):
img = depth_dataset[i,:,:]
#保存在一个文件夹：
cv2.imwrite("E:/data/{:0>5d}.jpg".format(i), img)
#保存在不同文件夹：
# if i < 100:
#
cv2.imwrite("E:/train/{:0>5d}.jpg".format(i), img)
# elif i < 200 :
#
cv2.imwrite("E:/val/{:0>5d}.jpg".format(i), img)
# else :
#
cv2.imwrite("E:/test/{:0>5d}.jpg".format(i), img)