代码拉取完成,页面将自动刷新
# create data and label for CK+
# 0=anger 1=disgust, 2=fear, 3=happy, 4=sadness, 5=surprise, 6=contempt
# contain 135,177,75,207,84,249,54 images
import csv
import os
import numpy as np
import h5py
import skimage.io
ck_path = 'CK+48'
anger_path = os.path.join(ck_path, 'anger')
disgust_path = os.path.join(ck_path, 'disgust')
fear_path = os.path.join(ck_path, 'fear')
happy_path = os.path.join(ck_path, 'happy')
sadness_path = os.path.join(ck_path, 'sadness')
surprise_path = os.path.join(ck_path, 'surprise')
contempt_path = os.path.join(ck_path, 'contempt')
# # Creat the list to store the data and label information
data_x = []
data_y = []
datapath = os.path.join('data','CK_data.h5')
if not os.path.exists(os.path.dirname(datapath)):
os.makedirs(os.path.dirname(datapath))
# order the file, so the training set will not contain the test set (don't random)
files = os.listdir(anger_path)
files.sort()
for filename in files:
I = skimage.io.imread(os.path.join(anger_path,filename))
data_x.append(I.tolist())
data_y.append(0)
files = os.listdir(disgust_path)
files.sort()
for filename in files:
I = skimage.io.imread(os.path.join(disgust_path,filename))
data_x.append(I.tolist())
data_y.append(1)
files = os.listdir(fear_path)
files.sort()
for filename in files:
I = skimage.io.imread(os.path.join(fear_path,filename))
data_x.append(I.tolist())
data_y.append(2)
files = os.listdir(happy_path)
files.sort()
for filename in files:
I = skimage.io.imread(os.path.join(happy_path,filename))
data_x.append(I.tolist())
data_y.append(3)
files = os.listdir(sadness_path)
files.sort()
for filename in files:
I = skimage.io.imread(os.path.join(sadness_path,filename))
data_x.append(I.tolist())
data_y.append(4)
files = os.listdir(surprise_path)
files.sort()
for filename in files:
I = skimage.io.imread(os.path.join(surprise_path,filename))
data_x.append(I.tolist())
data_y.append(5)
files = os.listdir(contempt_path)
files.sort()
for filename in files:
I = skimage.io.imread(os.path.join(contempt_path,filename))
data_x.append(I.tolist())
data_y.append(6)
print(np.shape(data_x))
print(np.shape(data_y))
datafile = h5py.File(datapath, 'w')
datafile.create_dataset("data_pixel", dtype = 'uint8', data=data_x)
datafile.create_dataset("data_label", dtype = 'int64', data=data_y)
datafile.close()
print("Save data finish!!!")
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。