将自制数据集转换成cifar10格式

def imread(im_path, shape=None, color="RGB"):
    im = cv2.imread(im_path, cv2.IMREAD_UNCHANGED)
    if color == "RGB":
        im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
    if shape != None:
        im = cv2.resize(im, shape)
    return im

c = 1024 # 32*32
# 3072 = 32*32*3
def default_loader(path):
    im = imread(path, shape=(32,32), color='RGB')
    data = np.zeros((3072,), dtype=np.uint8)
    # im[:,:,0].shape : (1024,)
    data[:c] = np.reshape(im[:,:,0],c)
    data[c:2*c] = np.reshape(im[:,:,1],c)
    data[2*c:] = np.reshape(im[:,:,2],c)

    # data:(3072,)
    return data

参考链接：手把手创建自己的数据集：类似cifar10数据结构的数据集_DYF-AI的博客-CSDN博客