将猫狗数据存为PICKLE文件，不随机初始化

# import cv2
# for pic in range(10):
#     for i in range(10):
#         image = cv2.imread('E:/catdog/data/train/dog/dog.'+str(pic)+'.jpg')
#         cv2.line(image, (60+i*10, 90), (60+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (61+i*10, 90), (61+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (62+i*10, 90), (62+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (63+i*10, 90), (63+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (64+i*10, 90), (64+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (65+i*10, 90), (65+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (66+i*10, 90), (66+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (67+i*10, 90), (67+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (68+i*10, 90), (68+i*10, 100), (0, 0, 0), 1)
#         cv2.line(image, (69+i*10, 90), (69+i*10, 100), (0, 0, 0), 1)
#         cv2.imwrite('E:/catdog/data/train/dog2/dog'+str(pic)+'.'+str(i)+'.jpg', image)

#加上zhedang的数据，在预处理这里直接就做了
import os
import cv2
from sklearn.utils import shuffle
import numpy as np
from six.moves import cPickle as pickle

CLASS_NAME = ["dog", "cat"]
img_size = 224
# validation = 0.1
train_path = 'E:/catdog/data/train/'
def get_dataset(path):
    images = []
    labels = []
    for i, name in enumerate(CLASS_NAME):
        direct_name = path + name + '/'
        direct = os.listdir(direct_name)
        for file in direct:
            img = cv2.imread(direct_name + file)
            img = cv2.resize(img, (img_size, img_size), 0, 0, cv2.INTER_LINEAR)
            img = img.astype(np.float32)
            img = np.multiply(img, 1.0 / 255.0)
            images.append(img)
            labels.append(i)
    images = np.array(images)
    labels = np.array(labels)
    return images, labels

def save_pickle(filename):
    # 获取数据
    train_images, train_labels = get_dataset(train_path)
    f = open(filename, 'wb')
    save = {
       'train_images': train_images,
       'train_label': train_labels,
    }
    pickle.dump(save, f, pickle.HIGHEST_PROTOCOL)
    f.close()

def main():
    save_pickle('D:/Animal.pickle')

if __name__ == '__main__':
    main()