import pickle
import copy
from PIL import Image

data = {}

dir_name = "./data/data04_train"
for i in range(1, 100001):
    file_name = f"{dir_name}/{i}.jpg"
    im = Image.open(file_name)
    data[f'{i}.jpg'] = copy.deepcopy(im)
    im.close()

pickle.dump(data, open('data.pkl', 'wb'))