import cv2
import numpy as np
file = './data/cifar-10-batches-py/test_batch'
# print(file)
def unpickle(file):
# import _pickle as cPickle
import pickle
with open(file, 'rb') as fo:
dict = pickle.load(fo,encoding="latin1")
return dict
dict1 = unpickle(file)
for i in range(10000):
img = dict1["data"][i]
img = np.reshape(img, (3, 32, 32))
img = img.transpose((1, 2, 0))
img_name = dict1["filenames"][i]
img_label = str(dict1["labels"][i])
my_path = "./data/png/test/" + img_label + "/" + img_name
# 这个是关键,相当于存到对应文件夹下了
cv2.imwrite(my_path, img)
cafir10分类图片转化为png图片(以文件夹分类好)
最新推荐文章于 2022-06-24 10:54:36 发布