保存pickle

import os
import cv2 as cv
import pickle
def read_data():
    path = 'C:/Users\lenovo\Desktop\shiyan_data/'
    train_img_1 = []
    train_img_2 = []
    train_label = []
    file_names = sorted(os.listdir(path))
    for file_name in file_names:
        if file_name[-4:].upper() == '.BMP':
            img = cv.imread(os.path.join(path, file_name))
        if 'gt.bmp' in file_name.lower():
            img = cv.cvtColor(img, cv.COLOR_RGB2GRAY)
            train_label.append(img)
        elif 'im1.bmp' in file_name.lower():
            train_img_1.append(img)
        elif 'im2.bmp' in file_name.lower():
            train_img_2.append(img)


    file = open("C:/Users\lenovo\Desktop\shiyan_data/train_sample_1.pickle", "wb")
    pickle.dump(train_img_1, file)
    file.close()
    file = open("C:/Users\lenovo\Desktop\shiyan_data/train_sample_2.pickle", "wb")
    pickle.dump(train_img_2, file)
    file.close()
    file = open("C:/Users\lenovo\Desktop\shiyan_data/train_label.pickle", "wb")
    pickle.dump(train_label, file)
    file.close()
    return train_img_1, train_img_2, train_label

read_data()