利用numpy ,opencv将图片数据,标签列表,转化为图片,并保存

利用numpy ,opencv将图片数据,标签列表,转化为图片,并保存

import osimport numpy as npimport cv2#图片数据列表 取值范围0-255data=[]#图片标签列表,label=[]#将数据列表转化成图片格式,-1 表示图片的数量,3表示图像的通道数,32表示图片的大小imgs = np.reshape(data, [-1, 3, 32, 32])for i in range(imgs.shape[0]): #获取第i张图片的数据 im_data = imgs[i, ...] #opencv的通道维度在后面,所以需要进行通道维度的跟换 im_data = np.transpose(im_data, [1, 2, 0]) #将rgb图片转化成bgr图片 im_data = cv2.cvtColor(im_data, cv2.COLOR_RGB2BGR) #根据不同的类别,生成不同类别的路径 f="{}/{}".format("data/image/train",classification[labels[i]]) if not os.path.exists(f): os.mkdir(f) #利用opencv写图片的方式生成图片,并按照图片的次序给图片命名 cv2.imwrite("{}/{}.jpg".format(f, str(i)), im_data)

本代码用在cifar10数据集上验证成功!!!

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。